Skip to content

Commit

Permalink
fix tests
Browse files Browse the repository at this point in the history
  • Loading branch information
bobokvsky committed Aug 19, 2024
1 parent fab83ca commit a458dce
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion datapipe/step/batch_transform.py
Original file line number Diff line number Diff line change
Expand Up @@ -642,7 +642,7 @@ def alter_res_df():
for df in pd.read_sql_query(u1, con=con, chunksize=chunk_size):
df = df[self.transform_keys]

if extra_filters is not None:
if extra_filters is not None and len(extra_filters) > 0:
df__extra_filters = pd.DataFrame(extra_filters)
if set(df__extra_filters.columns).intersection(df.columns):
df = pd.merge(df, df__extra_filters)
Expand Down

0 comments on commit a458dce

Please sign in to comment.