Benchmark pytorch: fix read config

Author: maximpavliv

dlclive/benchmark_pytorch.py

@@ -179,7 +179,7 @@ def analyze_video(
     frame_index = 0
 
     # Retrieve bodypart names and number of keypoints
-    bodyparts = dlc_live.cfg["metadata"]["bodyparts"]
+    bodyparts = dlc_live.read_config()["metadata"]["bodyparts"]
     num_keypoints = len(bodyparts)
 
     if save_video:

dlclive/core/runner.py

@@ -87,8 +87,14 @@ def init_inference(self, frame: np.ndarray | None, **kwargs) -> np.ndarray | Non
         pass
 
     @abc.abstractmethod
-    def read_config(self):
-        """Reads the pose configuration file.
+    def read_config(self) -> dict:
+        """
+        Reads the pose configuration file.
+
+        Returns
+        -------
+        dict
+            The runner configuration
 
         Raises:
             FileNotFoundError: if the pose configuration file does not exist

dlclive/dlclive.py

@@ -214,15 +214,20 @@ def cfg(self) -> dict | None:
     def precision(self) -> str:
         return self.runner.precision
 
-    def read_config(self) -> None:
+    def read_config(self) -> dict:
         """Reads configuration yaml file
 
+        Returns
+        -------
+        dict
+            The configuration
+
         Raises
         ------
         FileNotFoundError
             error thrown if pose configuration file does not exist
         """
-        self.runner.read_config()
+        return self.runner.read_config()
 
     @property
     def parameterization(