Merge branch 'jasmine-data2mobmat_accuracy_limit' of https://github.c…

…om/onnela-lab/forest into jasmine-data2mobmat_accuracy_limit
onnela-lab · Dec 15, 2023 · fc6c927 · fc6c927
2 parents c4630e5 + 6aaf70a
commit fc6c927
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/forest/jasmine/data2mobmat.py b/forest/jasmine/data2mobmat.py
@@ -1004,7 +1004,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)