code-loader 1.0.61.dev8__tar.gz → 1.0.61.dev10__tar.gz

{code_loader-1.0.61.dev8 → code_loader-1.0.61.dev10}/PKG-INFO

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

{code_loader-1.0.61.dev8 → code_loader-1.0.61.dev10}/code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -22,7 +22,7 @@ def tensorleap_custom_metric(name: str, direction: Optional[MetricDirection] = M
             ConfusionMatrixCallableInterfaceMultiArgs]
     ):
         for metric_handler in leap_binder.setup_container.metrics:
-            if metric_handler.name == name:
+            if metric_handler.metric_handler_data.name == name:
                 raise Exception(f'Metric with name {name} already exists. '
                                 f'Please choose another')
 
@@ -92,7 +92,7 @@ def tensorleap_custom_visualizer(name: str, visualizer_type: LeapDataType,
                                  heatmap_function: Optional[Callable[..., npt.NDArray[np.float32]]] = None):
     def decorating_function(user_function: VisualizerCallableInterface):
         for viz_handler in leap_binder.setup_container.visualizers:
-            if viz_handler.name == name:
+            if viz_handler.visualizer_handler_data.name == name:
                 raise Exception(f'Visualizer with name {name} already exists. '
                                 f'Please choose another')
 
@@ -328,7 +328,7 @@ def tensorleap_gt_encoder(name: str):
 def tensorleap_custom_loss(name: str):
     def decorating_function(user_function: CustomCallableInterface):
         for loss_handler in leap_binder.setup_container.custom_loss_handlers:
-            if loss_handler.name == name:
+            if loss_handler.custom_loss_handler_data.name == name:
                 raise Exception(f'Custom loss with name {name} already exists. '
                                 f'Please choose another')
 

{code_loader-1.0.61.dev8 → code_loader-1.0.61.dev10}/code_loader/leaploader.py

@@ -1,11 +1,12 @@
 # mypy: ignore-errors
 import importlib.util
+import inspect
 import io
 import sys
 from contextlib import redirect_stdout
 from functools import lru_cache
 from pathlib import Path
-from typing import Dict, List, Iterable, Union, Any, Type
+from typing import Dict, List, Iterable, Union, Any, Type, Optional
 
 import numpy as np
 import numpy.typing as npt
@@ -13,7 +14,7 @@ import numpy.typing as npt
 from code_loader.contract.datasetclasses import DatasetSample, DatasetBaseHandler, GroundTruthHandler, \
     PreprocessResponse, VisualizerHandler, LeapData, \
     PredictionTypeHandler, MetadataHandler, CustomLayerHandler, MetricHandler, VisualizerHandlerData, MetricHandlerData, \
-    MetricCallableReturnType, CustomLossHandlerData, CustomLossHandler
+    MetricCallableReturnType, CustomLossHandlerData, CustomLossHandler, RawInputsForHeatmap
 from code_loader.contract.enums import DataStateEnum, TestingSectionEnum, DataStateType, DatasetMetadataType
 from code_loader.contract.exceptions import DatasetScriptException
 from code_loader.contract.responsedataclasses import DatasetIntegParseResult, DatasetTestResultPayload, \
@@ -251,6 +252,16 @@ class LeapLoader(LeapLoaderBase):
             return list(input_tensors_by_arg_name.values())[0]
         return heatmap_function(**input_tensors_by_arg_name)
 
+    def get_heatmap_visualizer_raw_vis_input_arg_name(self, visualizer_name: str) -> Optional[str]:
+        heatmap_function = self._visualizer_handler_by_name()[visualizer_name].heatmap_function
+        if heatmap_function is None:
+            return None
+
+        for arg_name, arg_type in inspect.getfullargspec(heatmap_function).annotations.items():
+            if arg_type == RawInputsForHeatmap:
+                return arg_name
+        return None
+
     def get_dataset_setup_response(self, handlers_test_payloads: List[DatasetTestResultPayload]) -> DatasetSetup:
         setup = global_leap_binder.setup_container
         assert setup.preprocess is not None
@@ -429,3 +440,4 @@ class LeapLoader(LeapLoaderBase):
 
         return id_type
 
+

{code_loader-1.0.61.dev8 → code_loader-1.0.61.dev10}/code_loader/leaploaderbase.py

@@ -2,7 +2,7 @@
 
 from abc import abstractmethod
 
-from typing import Dict, List, Union, Type
+from typing import Dict, List, Union, Type, Optional
 
 import numpy as np
 import numpy.typing as npt
@@ -84,6 +84,10 @@ class LeapLoaderBase:
     def get_sample_id_type(self) -> Type:
         pass
 
+    @abstractmethod
+    def get_heatmap_visualizer_raw_vis_input_arg_name(self, visualizer_name: str) -> Optional[str]:
+        pass
+
     def is_custom_latent_space(self) -> bool:
         if not self.code_entry_name or not self.code_path:
             return False

{code_loader-1.0.61.dev8 → code_loader-1.0.61.dev10}/pyproject.toml

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