https://github.com/datajoint/element-deeplabcut/issues/82

SFB1089 · Jul 28, 2023 · 74b1f34 · 74b1f34
1 parent 7e535a8
commit 74b1f34
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/element_deeplabcut/model.py b/element_deeplabcut/model.py
@@ -776,7 +776,7 @@ def get_trajectory(cls, key: dict, body_parts: list = "all") -> pd.DataFrame:
         df = None
         for body_part in body_parts:
             x_pos, y_pos, z_pos, likelihood = (
-                cls.BodyPartPosition & {"body_part": body_part}
+                cls.BodyPartPosition & key & {"body_part": body_part} #TR23: added key restraint!
             ).fetch1("x_pos", "y_pos", "z_pos", "likelihood")
             if not z_pos:
                 z_pos = np.zeros_like(x_pos)