diff options
| -rw-r--r-- | src/dataframes.py | 4 | ||||
| -rw-r--r-- | src/transform_lambda.py | 3 |
2 files changed, 4 insertions, 3 deletions
diff --git a/src/dataframes.py b/src/dataframes.py index f2cae5d..d0479f1 100644 --- a/src/dataframes.py +++ b/src/dataframes.py @@ -115,9 +115,9 @@ def create_dim_location(dict_of_df): df_loc = ( dict_of_df["address"] .drop(labels=["created_at", "last_updated"], axis=1) - .rename(columns={"address_id": "location_id"}) + .rename(columns={"address_id": "location_id"})) return df_loc - + def create_dim_counterparty(dict_of_df): diff --git a/src/transform_lambda.py b/src/transform_lambda.py index 7677f66..57e9042 100644 --- a/src/transform_lambda.py +++ b/src/transform_lambda.py @@ -5,12 +5,13 @@ import logging import pandas as pd import pyarrow as pa import pyarrow.parquet as pq -from src.dataframes import * +from dataframes import * from botocore.exceptions import ClientError from pg8000.native import Connection, InterfaceError from datetime import datetime + class DBConnectionException(Exception): """Wraps pg8000.native Error or DatabaseError.""" |