code-loader 1.0.40a0__py3-none-any.whl → 1.0.40a1__py3-none-any.whl

code_loader/contract/datasetclasses.py

@@ -4,7 +4,7 @@ from typing import Any, Callable, List, Optional, Dict, Union, Type
 import numpy as np
 import numpy.typing as npt
 
-from code_loader.contract.enums import DataStateType, DataStateEnum, LeapDataType, ConfusionMatrixValue, MetricDirection
+from code_loader.contract.enums import DataStateType, DataStateEnum, LeapDataType, ConfusionMatrixValue, MetricDirection, InstanceAnalysisType
 from code_loader.contract.visualizer_classes import LeapImage, LeapText, LeapGraph, LeapHorizontalBar, \
     LeapTextMask, LeapImageMask, LeapImageWithBBox
 
@@ -121,7 +121,14 @@ class DatasetBaseHandler:
 @dataclass
 class InputHandler(DatasetBaseHandler):
     shape: Optional[List[int]] = None
-    instance_function: Optional[InstanceCallableInterface] = None
+
+
+@dataclass
+class ElementInstanceHandler:
+    input_name: str
+    instance_function: InstanceCallableInterface
+    analysis_type: InstanceAnalysisType
+
 
 
 @dataclass
@@ -157,6 +164,7 @@ class DatasetIntegrationSetup:
     unlabeled_data_preprocess: Optional[UnlabeledDataPreprocessHandler] = None
     visualizers: List[VisualizerHandler] = field(default_factory=list)
     inputs: List[InputHandler] = field(default_factory=list)
+    element_instances: List[ElementInstanceHandler] = field(default_factory=list)
     ground_truths: List[GroundTruthHandler] = field(default_factory=list)
     metadata: List[MetadataHandler] = field(default_factory=list)
     prediction_types: List[PredictionTypeHandler] = field(default_factory=list)

code_loader/contract/enums.py

@@ -63,3 +63,9 @@ class ConfusionMatrixValue(Enum):
 class TestingSectionEnum(Enum):
     Warnings = "Warnings"
     Errors = "Errors"
+
+
+
+class InstanceAnalysisType(Enum):
+    MaskInput = "MaskInput"
+    MaskLatentSpace = "MaskLatentSpace"

code_loader/inner_leap_binder/leapbinder.py

@@ -10,8 +10,8 @@ from code_loader.contract.datasetclasses import SectionCallableInterface, InputH
     MetadataSectionCallableInterface, UnlabeledDataPreprocessHandler, CustomLayerHandler, MetricHandler, \
     CustomCallableInterfaceMultiArgs, ConfusionMatrixCallableInterfaceMultiArgs, VisualizerCallableReturnType, \
     CustomMultipleReturnCallableInterfaceMultiArgs, DatasetBaseHandler, custom_latent_space_attribute, \
-    RawInputsForHeatmap, InstanceCallableInterface
-from code_loader.contract.enums import LeapDataType, DataStateEnum, DataStateType, MetricDirection
+    RawInputsForHeatmap, InstanceCallableInterface, ElementInstanceHandler
+from code_loader.contract.enums import LeapDataType, DataStateEnum, DataStateType, MetricDirection, InstanceAnalysisType
 from code_loader.contract.responsedataclasses import DatasetTestResultPayload
 from code_loader.contract.visualizer_classes import map_leap_data_type_to_visualizer_class
 from code_loader.utils import to_numpy_return_wrapper, get_shape
@@ -96,13 +96,16 @@ class LeapBinder:
     def set_unlabeled_data_preprocess(self, function: Callable[[], PreprocessResponse]) -> None:
         self.setup_container.unlabeled_data_preprocess = UnlabeledDataPreprocessHandler(function)
 
-    def set_input(self, function: SectionCallableInterface, name: str,
-                  instance_function: Optional[InstanceCallableInterface] = None) -> None:
+    def set_input(self, function: SectionCallableInterface, name: str) -> None:
         function = to_numpy_return_wrapper(function)
-        self.setup_container.inputs.append(InputHandler(name, function, instance_function=instance_function))
+        self.setup_container.inputs.append(InputHandler(name, function))
 
         self._encoder_names.append(name)
 
+    def set_instance_element(self, input_name: str, instance_function: Optional[InstanceCallableInterface] = None,
+                             analysis_type: InstanceAnalysisType = InstanceAnalysisType.MaskInput) -> None:
+        self.setup_container.element_instances.append(ElementInstanceHandler(input_name, instance_function, analysis_type))
+
     def add_custom_loss(self, function: CustomCallableInterface, name: str) -> None:
         arg_names = inspect.getfullargspec(function)[0]
         self.setup_container.custom_loss_handlers.append(CustomLossHandler(name, function, arg_names))

code_loader/leaploader.py

@@ -298,13 +298,15 @@ class LeapLoader:
     def _get_inputs(self, state: DataStateEnum, idx: int) -> Dict[str, npt.NDArray[np.float32]]:
         return self._get_dataset_handlers(global_leap_binder.setup_container.inputs, state, idx)
 
-    def get_instance_elements(self, state: DataStateEnum, idx: int, input_name: str) -> Optional[List[ElementInstance]]:
+    def get_instance_elements(self, state: DataStateEnum, idx: int, input_name: str) \
+            -> Tuple[Optional[List[ElementInstance]], Optional[InstanceAnalysisType]]:
         preprocess_result = self._preprocess_result()
         preprocess_state = preprocess_result[state]
-        for input in global_leap_binder.setup_container.inputs:
-            if input.name == input_name:
-                if input.instance_function is not None:
-                    return input.instance_function(idx, preprocess_state)
+        for element_instance in global_leap_binder.setup_container.element_instances:
+            if element_instance.input_name == input_name:
+                return element_instance.instance_function(idx, preprocess_state), element_instance.analysis_type
+
+        return None, None
 
 
     def _get_gt(self, state: DataStateEnum, idx: int) -> Dict[str, npt.NDArray[np.float32]]:

{code_loader-1.0.40a0.dist-info → code_loader-1.0.40a1.dist-info}/METADATA

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

{code_loader-1.0.40a0.dist-info → code_loader-1.0.40a1.dist-info}/RECORD

@@ -1,18 +1,18 @@
 LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
 code_loader/__init__.py,sha256=V3DEXSN6Ie6PlGeSAbzjp9ufRj0XPJLpD7pDLLYxk6M,122
 code_loader/contract/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-code_loader/contract/datasetclasses.py,sha256=eVPrNAoU3ViaxeZQof-qS2bWsgtuJXTqfCF9psS1f7k,4950
-code_loader/contract/enums.py,sha256=wJSMFXL_E-JWK_XAgsqOZwOLDiRfcT_rp0oQ3P_edNI,1560
+code_loader/contract/datasetclasses.py,sha256=45g-iy_HEd1Q7LAMaAjNbYqJxgEwvTlNNC_V6hN5koU,5141
+code_loader/contract/enums.py,sha256=3jBkDBa7od9SrSZdozrvy4InvyuRk4Ov1yEMsQxc-Pg,1665
 code_loader/contract/exceptions.py,sha256=jWqu5i7t-0IG0jGRsKF4DjJdrsdpJjIYpUkN1F4RiyQ,51
 code_loader/contract/responsedataclasses.py,sha256=WSHmFZWOFhGL1eED1u-aoRotPQg2owFQ-t3xSViWXSI,2808
 code_loader/contract/visualizer_classes.py,sha256=1FjVO744J_EMuJfHWXGdvSz6vl3Vu7iS3CDfs8MzEEQ,5138
 code_loader/inner_leap_binder/__init__.py,sha256=koOlJyMNYzGbEsoIbXathSmQ-L38N_pEXH_HvL7beXU,99
-code_loader/inner_leap_binder/leapbinder.py,sha256=n9oiuC-bFLQsG-ZRSO6VeOWg2CB6qHZciAjRlIFKmi8,13778
-code_loader/leaploader.py,sha256=hsU1SdUTZIU_sRMGs9LjxMuvIECKsKWq_wL9W5sKMX4,17780
+code_loader/inner_leap_binder/leapbinder.py,sha256=wf3yxyxhN8k_eFecmt7nTUqbD3uMEoGccZcrOilOftc,14054
+code_loader/leaploader.py,sha256=D1CzclYll0kg_2sTZ_tS1-BFWlMCmiJZHrQCMwRiNtg,17882
 code_loader/utils.py,sha256=61I4PgSl-ZBIe4DifLxMNlBELE-HQR2pB9efVYPceIU,2230
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=HqWx2qfTrroGl2n8Fpmr_4X-rk7tE2oGapjO3gzz4WY,2226
-code_loader-1.0.40a0.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.40a0.dist-info/METADATA,sha256=ZZDeEYKHHQLlyqMKFzGXyKpWqui6gg9HWXcWrU6FdSs,770
-code_loader-1.0.40a0.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-code_loader-1.0.40a0.dist-info/RECORD,,
+code_loader-1.0.40a1.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.40a1.dist-info/METADATA,sha256=o9NkQbpAF3lZIRPc147tORbE0NHH1gBJ7igKxKvDCIg,770
+code_loader-1.0.40a1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+code_loader-1.0.40a1.dist-info/RECORD,,

{code_loader-1.0.40a0.dist-info → code_loader-1.0.40a1.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.8.1
+Generator: poetry-core 1.9.0
 Root-Is-Purelib: true
 Tag: py3-none-any