fix: specify str dtypes reading acdc_df csv

SchmollerLab · May 7, 2024 · ef1d998 · ef1d998
1 parent 846b40a
commit ef1d998
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/cellacdc/workers.py b/cellacdc/workers.py
@@ -695,8 +695,9 @@ def _save_acdc_df(self, recovery_acdc_df, posData):
             load.store_unsaved_acdc_df(recovery_folderpath, recovery_acdc_df)
             return
 
+        saved_acdc_df_path = posData.acdc_output_csv_path
         saved_acdc_df = (
-            pd.read_csv(posData.acdc_output_csv_path)
+            pd.read_csv(saved_acdc_df_path, dtype=load.acdc_df_str_cols)
             .set_index(['frame_i', 'Cell_ID'])
         )
 
@@ -721,7 +722,7 @@ def _save_acdc_df(self, recovery_acdc_df, posData):
             reference_acdc_df = saved_acdc_df
         else:
             reference_acdc_df = (
-                pd.read_csv(last_unsaved_csv_path)
+                pd.read_csv(last_unsaved_csv_path, dtype=load.acdc_df_str_cols)
                 .set_index(['frame_i', 'Cell_ID'])
             )