code-loader 1.0.62__tar.gz → 1.0.64a0__tar.gz

{code_loader-1.0.62 → code_loader-1.0.64a0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: code-loader
-Version: 1.0.62
+Version: 1.0.64a0
 Summary: 
 Home-page: https://github.com/tensorleap/code-loader
 License: MIT
@@ -13,7 +13,6 @@ Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
-Requires-Dist: matplotlib (>=3.3,<3.4)
 Requires-Dist: numpy (>=1.22.3,<2.0.0)
 Requires-Dist: psutil (>=5.9.5,<6.0.0)
 Requires-Dist: pyyaml (>=6.0.2,<7.0.0)

{code_loader-1.0.62 → code_loader-1.0.64a0}/code_loader/contract/datasetclasses.py

@@ -181,6 +181,7 @@ class CustomLayerHandler:
     init_arg_names: List[str]
     call_arg_names: List[str]
     use_custom_latent_space: bool = False
+    build: bool = True
 
 
 @dataclass

{code_loader-1.0.62 → code_loader-1.0.64a0}/code_loader/inner_leap_binder/leapbinder.py

@@ -339,7 +339,8 @@ class LeapBinder:
         self.setup_container.metadata.append(MetadataHandler(name, function))
 
     def set_custom_layer(self, custom_layer: Type[Any], name: str, inspect_layer: bool = False,
-                         kernel_index: Optional[int] = None, use_custom_latent_space: bool = False) -> None:
+                         kernel_index: Optional[int] = None, use_custom_latent_space: bool = False,
+                         build=True) -> None:
         """
         Set a custom layer for the model.
 
@@ -369,7 +370,8 @@ class LeapBinder:
         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,
-                                                                      use_custom_latent_space=use_custom_latent_space)
+                                                                      use_custom_latent_space=use_custom_latent_space,
+                                                                      build=build)
 
     def check_preprocess(self, preprocess_result: Dict[DataStateEnum, PreprocessResponse]) -> None:
         preprocess_handler = self.setup_container.preprocess

{code_loader-1.0.62 → code_loader-1.0.64a0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "code-loader"
-version = "1.0.62"
+version = "1.0.64a0"
 description = ""
 authors = ["dorhar <doron.harnoy@tensorleap.ai>"]
 license = "MIT"
@@ -15,7 +15,6 @@ include = [
 python = ">=3.8,<3.12"
 numpy = "^1.22.3"
 psutil = "^5.9.5"
-matplotlib = ">=3.3,<3.4"
 requests = "^2.32.3"
 pyyaml = "^6.0.2"