Add MetricDict class

Author: fcogidi

cyclops/evaluate/metrics/experimental/__init__.py

@@ -17,6 +17,7 @@
     MultilabelF1Score,
     MultilabelFBetaScore,
 )
+from cyclops.evaluate.metrics.experimental.metric_dict import MetricDict
 from cyclops.evaluate.metrics.experimental.precision_recall import (
     BinaryPrecision,
     BinaryRecall,

cyclops/evaluate/metrics/experimental/metric.py

@@ -488,6 +488,19 @@ def __deepcopy__(self, memo: Optional[Dict[int, Any]] = None) -> "Metric":
                     lambda x: apc.to_device(clone(x), self.device),
                 )
                 setattr(obj_copy, k, _defaults_)
+            elif isinstance(v, (list, tuple)):
+                seq_var = apply_to_array_collection(
+                    v,
+                    lambda x: apc.to_device(clone(x), self.device),
+                )
+                setattr(
+                    obj_copy,
+                    k,
+                    [
+                        deepcopy(arr, memo) if not apc.is_array_api_obj(arr) else arr
+                        for arr in seq_var
+                    ],
+                )
             elif apc.is_array_api_obj(v):
                 setattr(obj_copy, k, apc.to_device(clone(v), self.device))
             else: