code-loader 1.0.22__tar.gz → 1.0.23__tar.gz

{code_loader-1.0.22 → code_loader-1.0.23}/PKG-INFO

@@ -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 → code_loader-1.0.23}/code_loader/contract/datasetclasses.py

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

{code_loader-1.0.22 → code_loader-1.0.23}/code_loader/contract/responsedataclasses.py

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

{code_loader-1.0.22 → code_loader-1.0.23}/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-1.0.22 → code_loader-1.0.23}/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 → code_loader-1.0.23}/pyproject.toml

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