Fix warning about concatenating empty or NA data frames

Project-Platypus · Sep 6, 2024 · 5150f3d · 5150f3d
1 parent 2e94224
commit 5150f3d
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 10 deletions.
diff --git a/prim/prim_box.py b/prim/prim_box.py
@@ -404,16 +404,16 @@ def update(self, box_lims, indices):
         self._box_lims.append(box_lims)
         coi = self.prim.determine_coi(self.yi)
 
-        stats = {"coverage" : coi/self.prim.t_coi, 
-                "density" : coi/y.shape[0],  
-                "mean" : np.mean(y),
-                "res dim" : determine_nr_restricted_dims(
-                        self._box_lims[-1], 
-                        self.prim._box_init),
-                "mass" : y.shape[0]/self.prim.n}
-
-        self.peeling_trajectory = pd.concat([self.peeling_trajectory, pd.DataFrame([stats])], 
-                ignore_index=True)
+        print(self.prim.t_coi, y.shape[0], self.prim.n)
+
+        stats = pd.DataFrame([{"coverage": coi/self.prim.t_coi, 
+                               "density": coi/y.shape[0],  
+                               "mean": np.mean(y),
+                               "res dim": determine_nr_restricted_dims(self._box_lims[-1], self.prim._box_init),
+                               "mass": y.shape[0]/self.prim.n}])
+
+        self.peeling_trajectory = stats if self.peeling_trajectory.empty else \
+            pd.concat([self.peeling_trajectory, stats], ignore_index=True)
 
         self._cur_box = len(self.peeling_trajectory)-1
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -29,3 +29,7 @@ test = ["pytest", "mock"]
 
 [tool.setuptools.dynamic]
 version = {attr = "prim.__version__"}
+
+[tool.pytest.ini_options]
+addopts = "-rA -Werror"
+testpaths = ["prim"]