code-loader 1.0.64a0__tar.gz → 1.0.64a1__tar.gz

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

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

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

@@ -181,7 +181,6 @@ 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.64a0 → code_loader-1.0.64a1}/code_loader/contract/visualizer_classes.py

@@ -1,4 +1,4 @@
-from typing import List, Any, Union
+from typing import List, Any, Union, Optional
 
 import numpy as np
 import numpy.typing as npt
@@ -112,11 +112,12 @@ class LeapText:
 
     Example:
         text_data = ['I', 'ate', 'a', 'banana', '', '', '']
-        leap_text = LeapText(data=text_data)  # Create LeapText object
-        LeapText(leap_text)
+        heatmap = [0.1, 0.3, 0.2, 0.9, 0.0, 0.0, 0.0]
+        leap_text = LeapText(data=text_data heatmap=heatmap)  # Create LeapText object
     """
     data: List[str]
     type: LeapDataType = LeapDataType.Text
+    heatmap: Optional[List[float]] = None
 
     def __post_init__(self) -> None:
         validate_type(self.type, LeapDataType.Text)
@@ -124,6 +125,15 @@ class LeapText:
         for value in self.data:
             validate_type(type(value), str)
 
+        if self.heatmap is not None:
+            validate_type(type(self.heatmap), list)
+            for value in self.heatmap:
+                validate_type(type(value), float)
+            if len(self.heatmap) != len(self.data):
+                raise LeapValidationError(
+                    f"Heatmap length ({len(self.heatmap)}) must match the number of tokens in `data` ({len(self.data)})."
+                )
+
 
 @dataclass
 class LeapHorizontalBar:

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

@@ -339,8 +339,7 @@ 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,
-                         build=True) -> None:
+                         kernel_index: Optional[int] = None, use_custom_latent_space: bool = False) -> None:
         """
         Set a custom layer for the model.
 
@@ -370,8 +369,7 @@ 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,
-                                                                      build=build)
+                                                                      use_custom_latent_space=use_custom_latent_space)
 
     def check_preprocess(self, preprocess_result: Dict[DataStateEnum, PreprocessResponse]) -> None:
         preprocess_handler = self.setup_container.preprocess

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

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