code-loader 1.0.171__tar.gz → 1.0.172.dev1__tar.gz

{code_loader-1.0.171 → code_loader-1.0.172.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: code-loader
-Version: 1.0.171
+Version: 1.0.172.dev1
 Summary: 
 Home-page: https://github.com/tensorleap/code-loader
 License: MIT

{code_loader-1.0.171 → code_loader-1.0.172.dev1}/code_loader/contract/datasetclasses.py

@@ -1,6 +1,6 @@
 import warnings
 from dataclasses import dataclass, field
-from typing import Any, Callable, List, Optional, Dict, Union, Type, Literal
+from typing import Any, Callable, Iterable, List, Optional, Dict, Union, Type, Literal, Tuple
 import re
 import numpy as np
 import numpy.typing as npt
@@ -165,6 +165,36 @@ class MetricHandlerData:
     compute_insights: Optional[Union[bool, Dict[str, bool]]] = None
 
 
+def normalize_metric_handler_settings(
+        direction: Union[None, MetricDirection, Dict[str, MetricDirection]],
+        compute_insights: Optional[Union[bool, Dict[str, bool]]],
+        result_keys: Iterable[str],
+) -> Tuple[Union[None, MetricDirection, Dict[str, MetricDirection]], Optional[Union[bool, Dict[str, bool]]]]:
+    ordered_result_keys = list(result_keys)
+    if not ordered_result_keys:
+        return direction, compute_insights
+
+    normalized_direction = direction
+    if isinstance(direction, MetricDirection):
+        normalized_direction = {key: direction for key in ordered_result_keys}
+    elif isinstance(direction, dict):
+        normalized_direction = {
+            key: direction.get(key, MetricDirection.Downward)
+            for key in ordered_result_keys
+        }
+
+    normalized_compute_insights = compute_insights
+    if isinstance(compute_insights, bool):
+        normalized_compute_insights = {key: compute_insights for key in ordered_result_keys}
+    elif isinstance(compute_insights, dict):
+        normalized_compute_insights = {
+            key: compute_insights.get(key, True)
+            for key in ordered_result_keys
+        }
+
+    return normalized_direction, normalized_compute_insights
+
+
 @dataclass
 class MetricHandler:
     metric_handler_data: MetricHandlerData
@@ -272,4 +302,3 @@ class DatasetSample:
     state: DataStateEnum
     custom_latent_space: Optional[npt.NDArray[np.float32]] = None
     instance_masks: Optional[Dict[str, ElementInstance]] = None
-

{code_loader-1.0.171 → code_loader-1.0.172.dev1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "code-loader"
-version = "1.0.171"
+version = "1.0.172.dev1"
 description = ""
 authors = ["dorhar <doron.harnoy@tensorleap.ai>"]
 license = "MIT"