From 74b1f34f8dd2fcbd8553944dfc0f15f11b7d027c Mon Sep 17 00:00:00 2001
From: Tobias Rose <tobias.rose@ukbonn.de>
Date: Fri, 28 Jul 2023 10:55:47 +0000
Subject: [PATCH] https://github.com/datajoint/element-deeplabcut/issues/82

---
 element_deeplabcut/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/element_deeplabcut/model.py b/element_deeplabcut/model.py
index 09269f8..d4faa56 100644
--- 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)