Address deprecation warning for pd.read_csv squeeze arg

wsp-sag · Feb 25, 2022 · 18c2e0d · 18c2e0d
1 parent 857c52a
commit 18c2e0d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/balsa/routines/io/nwp.py b/balsa/routines/io/nwp.py
@@ -359,7 +359,7 @@ def read_nwp_transit_station_results(nwp_fp: Union[str, Path], station_line_node
         raise FileNotFoundError(f'File `{nwp_fp.as_posix()}` not found.')
 
     with zipfile.ZipFile(nwp_fp) as zf:
-        results = pd.read_csv(zf.open('aux_transit_results.csv'), index_col=['i', 'j'], squeeze=True)
+        results = pd.read_csv(zf.open('aux_transit_results.csv'), index_col=['i', 'j']).squeeze('columns')
 
     station_results = pd.DataFrame(index=sorted(station_line_nodes))
     station_results.index.name = 'stn_node'