Merge branch 'develop' into jasmine-empty_flight_table

onnela-lab · Dec 15, 2023 · 58ca4cc · 58ca4cc
2 parents 1c127d8 + 07b2702
commit 58ca4cc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/forest/jasmine/data2mobmat.py b/forest/jasmine/data2mobmat.py
@@ -1000,7 +1000,8 @@ def infer_mobmat(mobmat: np.ndarray, interval: float, r: float) -> np.ndarray:
         (mobmat, np.ones(n_rows).reshape(n_rows, 1))
     )
     # Append new pauses to the trajectory matrix
-    mobmat = np.vstack((mobmat, new_pauses_array))
+    if new_pauses_array.shape[0] > 0:
+        mobmat = np.vstack((mobmat, new_pauses_array))
     # Sort the matrix by start time
     mobmat = mobmat[mobmat[:, 3].argsort()].astype(float)