Fixed bug on rule_based annotation

deepof/data.py

@@ -651,7 +651,8 @@ class coordinates:
         speeds = self.get_coords(speed=1)
         for key in tqdm(self._tables.keys()):
 
-            video = [vid for vid in self._videos if key in vid][0]
+            video = [vid for vid in self._videos if key + "DLC" in vid][0]
+            print(key, video)
             tag_dict[key] = deepof.pose_utils.rule_based_tagging(
                 list(self._tables.keys()),
                 self._videos,

deepof/pose_utils.py

@@ -136,7 +136,6 @@ def climb_wall(
     if arena_type == "circular":
         center = np.zeros(2) if centered_data else np.array(arena[:2])
         radius = arena[2]
-        print(radius)
         climbing = np.linalg.norm(nose - center, axis=1) > (radius + tol)
 
     else: