Skip to content

Commit

Permalink
Update error message
Browse files Browse the repository at this point in the history
  • Loading branch information
sundy1994 committed Feb 28, 2025
1 parent b50ea77 commit 253286c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/team_comm_tools/utils/preprocess.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ def assert_key_columns_present(df: pd.DataFrame, column_names: dict) -> None:
if role == 'timestamp_col':
continue # skip timestamp column
if col not in df.columns:
raise KeyError(f"Missing required columns in DataFrame: '{col}' (expected for {role})")
raise KeyError(f"Missing required columns in DataFrame: '{col}' (expected for {role})\n Columns available: {df.columns}")
else:
print(f"Confirmed that data has {role} column: {col}!")

Expand Down

0 comments on commit 253286c

Please sign in to comment.