code-loader 1.0.21__py3-none-any.whl → 1.0.23__py3-none-any.whl

code_loader/contract/datasetclasses.py

@@ -122,6 +122,7 @@ class MetadataHandler:
 class PredictionTypeHandler:
     name: str
     labels: List[str]
+    channel_dim: int
 
 
 @dataclass

code_loader/contract/responsedataclasses.py

@@ -62,6 +62,7 @@ class CustomLayerInstance:
 class PredictionTypeInstance:
     name: str
     labels: List[str]
+    channel_dim: int
 
 
 @dataclass

code_loader/inner_leap_binder/leapbinder.py

@@ -3,7 +3,6 @@ from typing import Callable, List, Optional, Dict, Any, Type, Union
 
 import numpy as np
 import numpy.typing as npt
-from typeguard import typechecked
 
 from code_loader.contract.datasetclasses import SectionCallableInterface, InputHandler, \
     GroundTruthHandler, MetadataHandler, DatasetIntegrationSetup, VisualizerHandler, PreprocessResponse, \
@@ -45,7 +44,6 @@ class LeapBinder:
         self.set_visualizer(function=default_text_mask_visualizer, name=DefaultVisualizer.TextMask.value,
                             visualizer_type=LeapDataType.TextMask)
 
-    @typechecked
     def set_visualizer(self, function: VisualizerCallableInterface,
                        name: str,
                        visualizer_type: LeapDataType,
@@ -85,27 +83,22 @@ class LeapBinder:
             VisualizerHandler(name, function, visualizer_type, arg_names, heatmap_visualizer))
         self._visualizer_names.append(name)
 
-    @typechecked
     def set_preprocess(self, function: Callable[[], List[PreprocessResponse]]) -> None:
         self.setup_container.preprocess = PreprocessHandler(function)
 
-    @typechecked
     def set_unlabeled_data_preprocess(self, function: Callable[[], PreprocessResponse]) -> None:
         self.setup_container.unlabeled_data_preprocess = UnlabeledDataPreprocessHandler(function)
 
-    @typechecked
     def set_input(self, function: SectionCallableInterface, name: str) -> None:
         function = to_numpy_return_wrapper(function)
         self.setup_container.inputs.append(InputHandler(name, function))
 
         self._encoder_names.append(name)
 
-    @typechecked
     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))
 
-    @typechecked
     def add_custom_metric(self,
                           function: Union[CustomCallableInterfaceMultiArgs,
                           CustomMultipleReturnCallableInterfaceMultiArgs,
@@ -114,22 +107,18 @@ class LeapBinder:
         arg_names = inspect.getfullargspec(function)[0]
         self.setup_container.metrics.append(MetricHandler(name, function, arg_names))
 
-    @typechecked
-    def add_prediction(self, name: str, labels: List[str]) -> None:
-        self.setup_container.prediction_types.append(PredictionTypeHandler(name, labels))
+    def add_prediction(self, name: str, labels: List[str], channel_dim=-1) -> None:
+        self.setup_container.prediction_types.append(PredictionTypeHandler(name, labels, channel_dim))
 
-    @typechecked
     def set_ground_truth(self, function: SectionCallableInterface, name: str) -> None:
         function = to_numpy_return_wrapper(function)
         self.setup_container.ground_truths.append(GroundTruthHandler(name, function))
 
         self._encoder_names.append(name)
 
-    @typechecked
     def set_metadata(self, function: MetadataSectionCallableInterface, name: str) -> None:
         self.setup_container.metadata.append(MetadataHandler(name, function))
 
-    @typechecked
     def set_custom_layer(self, custom_layer: Type[Any], name: str, inspect_layer: bool = False,
                          kernel_index: Optional[int] = None) -> None:
         if inspect_layer and kernel_index is not None:

code_loader/leaploader.py

@@ -256,7 +256,8 @@ class LeapLoader:
 
         prediction_types = []
         for prediction_type in setup.prediction_types:
-            pred_type_inst = PredictionTypeInstance(prediction_type.name, prediction_type.labels)
+            pred_type_inst = PredictionTypeInstance(prediction_type.name, prediction_type.labels,
+                                                    prediction_type.channel_dim)
             prediction_types.append(pred_type_inst)
 
         metrics = []

{code_loader-1.0.21.dist-info → code_loader-1.0.23.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: code-loader
-Version: 1.0.21
+Version: 1.0.23
 Summary: 
 Home-page: https://github.com/tensorleap/code-loader
 License: MIT
@@ -15,7 +15,6 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Requires-Dist: numpy (>=1.22.3,<2.0.0)
 Requires-Dist: psutil (>=5.9.5,<6.0.0)
-Requires-Dist: typeguard (>=4.1.5,<5.0.0)
 Project-URL: Repository, https://github.com/tensorleap/code-loader
 Description-Content-Type: text/markdown
 

{code_loader-1.0.21.dist-info → code_loader-1.0.23.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=h02sNsg4p-ypf5Dh00lIe1D3Bw1tnVMbKs2yO8X-2Vg,4433
+code_loader/contract/datasetclasses.py,sha256=xuPrL3cKZ-tNBMQ5FsSh7oI8U6ITA_QPI57LN2T8aF8,4454
 code_loader/contract/enums.py,sha256=KDIeNl79e8rCp4ybJat7U03j50McrcBTWUxko5SXrug,1481
 code_loader/contract/exceptions.py,sha256=jWqu5i7t-0IG0jGRsKF4DjJdrsdpJjIYpUkN1F4RiyQ,51
-code_loader/contract/responsedataclasses.py,sha256=thIxKSx3mwoKHYICULGaeh2-S56L1xBlmDThCE11yT4,2658
+code_loader/contract/responsedataclasses.py,sha256=0bo4iLbfu6-3tpgzeKakIH2i_1jtR8BiLdupa1TVT0o,2679
 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=oFpWLket1zqdvhJ1wWsERCtIkADQUPsDL26AuUfzSLI,12708
-code_loader/leaploader.py,sha256=eyCxajK7hdkFQHRhmXUjppT1BJ4dRcTRmGXxeHiaZKQ,15322
+code_loader/inner_leap_binder/leapbinder.py,sha256=C3276VCZaeOQ4WZnSB0aB9N3o_HqZgq-NXjViieJR10,12533
+code_loader/leaploader.py,sha256=XUziFXq89AFbqUoWnlh5sYkoGA8jqUziORcOQA6W0GQ,15403
 code_loader/utils.py,sha256=lluWgGpG7sYB-o9Wk9vZlm4uQqb7qbPZBa_qnSe0BkU,1933
 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.21.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.21.dist-info/METADATA,sha256=96ga8rPCjW1oriXIBhqn53oxOAvVFawNHr7Xpa0Mgvo,810
-code_loader-1.0.21.dist-info/WHEEL,sha256=d2fvjOD7sXsVzChCqf0Ty0JbHKBaLYwDbGQDwQTnJ50,88
-code_loader-1.0.21.dist-info/RECORD,,
+code_loader-1.0.23.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.23.dist-info/METADATA,sha256=wMwTiGzMTNC6jcLy2AKsv4etK70592Uhscajkr2zGic,768
+code_loader-1.0.23.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+code_loader-1.0.23.dist-info/RECORD,,

{code_loader-1.0.21.dist-info → code_loader-1.0.23.dist-info}/WHEEL

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