bugfix to deal with None entries

MickaelRigault · Dec 22, 2022 · 4ba98ec · 4ba98ec
1 parent 8536099
commit 4ba98ec
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/ztflc/forcephotometry.py b/ztflc/forcephotometry.py
@@ -154,7 +154,9 @@ def run_forcefit(
                 if bar is not None:
                     bar.update(index_count)
 
-            self._data_forcefit = pd.DataFrame(dataout_list)
+            dataout_final = [i for i in dataout_list if i is not None]
+
+            self._data_forcefit = pd.DataFrame(dataout_final)
             if store:
                 self.store()