Skip to content
Snippets Groups Projects
Commit d0ecaf1c authored by Nicolas Schmid's avatar Nicolas Schmid
Browse files

fix: check if dataframe empty before dropping column

parent ef3542a8
No related branches found
No related tags found
1 merge request!44fix: check if dataframe empty before dropping column
Pipeline #3232 passed
...@@ -167,7 +167,10 @@ async def get_section_hydraulics(borehole_id: uuid.UUID, ...@@ -167,7 +167,10 @@ async def get_section_hydraulics(borehole_id: uuid.UUID,
db_result_df = await crud.read_hydraulics_df( db_result_df = await crud.read_hydraulics_df(
section_oid, starttime, endtime, defer_cols) section_oid, starttime, endtime, defer_cols)
db_result_df = db_result_df.dropna(axis=1, how='all').drop(columns=['_oid']) db_result_df = db_result_df.dropna(axis=1, how='all')
if '_oid' in db_result_df.columns:
db_result_df = db_result_df.drop(columns=['_oid'])
if format == 'csv': if format == 'csv':
return csv_response(db_result_df) return csv_response(db_result_df)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment