code-loader 1.0.22__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

@@ -107,8 +107,8 @@ class LeapBinder:
         arg_names = inspect.getfullargspec(function)[0]
         self.setup_container.metrics.append(MetricHandler(name, function, arg_names))
 
-    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))
 
     def set_ground_truth(self, function: SectionCallableInterface, name: str) -> None:
         function = to_numpy_return_wrapper(function)
@@ -119,7 +119,11 @@ class LeapBinder:
     def set_metadata(self, function: MetadataSectionCallableInterface, name: str) -> None:
         self.setup_container.metadata.append(MetadataHandler(name, function))
 
-    def set_custom_layer(self, custom_layer: Type[Any], name: str) -> None:
+    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:
+            custom_layer.kernel_index = kernel_index
+
         init_args = inspect.getfullargspec(custom_layer.__init__)[0][1:]
         call_args = inspect.getfullargspec(custom_layer.call)[0][1:]
         self.setup_container.custom_layers[name] = CustomLayerHandler(name, custom_layer, init_args, call_args)

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.22.dist-info → code_loader-1.0.23.dist-info}/METADATA

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

{code_loader-1.0.22.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=ctlz4D75L8LJjCEBr9TcdWi2RrFMCiLN4UmRclmvauc,12305
-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.22.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.22.dist-info/METADATA,sha256=65xWzMqYH8T-ywVNs_hBbxUyeT6-wLwBsD9lJ3VHejM,768
-code_loader-1.0.22.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-code_loader-1.0.22.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,,