Move around metrics

Author: spencer@primus

avapi/_dataset.py

@@ -277,24 +277,25 @@ def get_objects(
         timestamp = self.get_timestamp(frame=frame, sensor=sensor, agent=agent)
         if max_occ is not None:
             objs = [
-                    obj
-                    for obj in objs
-                    if (obj.occlusion <= max_occ)
-                    or (obj.occlusion == Occlusion.UNKNOWN)
+                obj
+                for obj in objs
+                if (obj.occlusion <= max_occ) or (obj.occlusion == Occlusion.UNKNOWN)
             ]
         if max_dist is not None:
             if sensor == "ego":
                 calib = calibration.Calibration(reference)
             else:
                 calib = self.get_calibration(frame, sensor, agent=agent)
             objs = [
-                    obj
-                    for obj in objs
-                    if obj.position.distance(calib.reference) < max_dist
+                obj for obj in objs if obj.position.distance(calib.reference) < max_dist
             ]
-        objs = DataContainer(source_identifier=sensor, frame=frame, timestamp=timestamp, data=objs)
+        objs = DataContainer(
+            source_identifier=sensor, frame=frame, timestamp=timestamp, data=objs
+        )
         if in_global:
-            objs = objs.apply_and_return("change_reference", GlobalOrigin3D, inplace=False)
+            objs = objs.apply_and_return(
+                "change_reference", GlobalOrigin3D, inplace=False
+            )
         return objs
 
     def get_objects_global(

avapi/carla/dataset.py

@@ -277,7 +277,7 @@ def get_sensor_file(self, frame, timestamp, sensor, agent, file_type):
     def get_sensor_name(self, sensor, agent):
         return sensor
 
-    def get_agents(self, frame: int):
+    def get_agents(self, frame: int) -> "DataContainer":
         return self._load_agents(frame)
 
     def get_agent(self, frame: int, agent: int):

avapi/evaluation/__init__.py

@@ -1,4 +1,3 @@
-from . import metrics
 from .base import ResultManager
 from .perception import (
     get_percep_results_from_folder,
@@ -7,5 +6,3 @@
 from .prediction import get_predict_results_from_folder
 from .tracking import get_track_results_from_folder, get_track_results_from_multi_folder
 from .trades import run_trades
-
-

avapi/evaluation/base.py

@@ -16,7 +16,9 @@
 from avapi.utils import color_from_object_type, get_indices_filenames_in_folder
 
 from ..visualize import snapshot
-from .metrics import precision, recall
+
+
+# from .metrics.assignment import precision, recall
 
 
 # =============================================