From 72ebda950c84d7b519db9a236b35a7fafcbb1899 Mon Sep 17 00:00:00 2001 From: lian-manonog Date: Tue, 27 Aug 2024 09:30:34 +0100 Subject: wip: added a bracket in dataframes --- src/dataframes.py | 4 ++-- 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.""" -- cgit v1.2.3