code-loader 1.0.64.dev3__py3-none-any.whl → 1.0.65__py3-none-any.whl

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 v in self.heatmap:
+                validate_type(type(v), 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/default_metrics.py

@@ -1,3 +1,5 @@
+# mypy: ignore-errors
+
 from enum import Enum
 from typing import List, Tuple
 import numpy as np

code_loader/visualizers/default_visualizers.py

@@ -33,6 +33,8 @@ def default_horizontal_bar_visualizer(data: npt.NDArray[np.float32]) -> LeapHori
 
 
 def default_word_visualizer(data: npt.NDArray[np.float32]) -> LeapText:
+    if len(data.shape) == 2 and data.shape[0] == 1:
+        data = data[0]
     if hasattr(data, 'tolist'):
         data = data.tolist()
     words = [str(index[0]) if type(index) is list else str(index) for index in data]

{code_loader-1.0.64.dev3.dist-info → code_loader-1.0.65.dist-info}/METADATA

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

{code_loader-1.0.64.dev3.dist-info → code_loader-1.0.65.dist-info}/RECORD

@@ -6,8 +6,8 @@ code_loader/contract/datasetclasses.py,sha256=L_fSdSvf-eKoez2uBJ8VjfrKedEP0szNOP
 code_loader/contract/enums.py,sha256=6Lo7p5CUog68Fd31bCozIuOgIp_IhSiPqWWph2k3OGU,1602
 code_loader/contract/exceptions.py,sha256=jWqu5i7t-0IG0jGRsKF4DjJdrsdpJjIYpUkN1F4RiyQ,51
 code_loader/contract/responsedataclasses.py,sha256=RSx9m_R3LawhK5o1nAcO3hfp2F9oJYtxZr_bpP3bTmw,4005
-code_loader/contract/visualizer_classes.py,sha256=iIa_O2rKvPTwN5ILCTZvRpsGYiiFABKdwQwfIXGigDo,11928
-code_loader/default_metrics.py,sha256=2kSaB71OrbQIXlcTMStYUebL8N-8bE9se3m-AXVdvCY,3936
+code_loader/contract/visualizer_classes.py,sha256=zD7SVgI1R_DaGby3FPJY2rFphK162P7F2hRTHp2xUuM,12428
+code_loader/default_metrics.py,sha256=2LLOHyeHhAj_f_l76hV7aNr2e9AWusCL48OMxln_o5g,3959
 code_loader/experiment_api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/experiment_api/api.py,sha256=a7wh6Hhe7IaVxu46eV2soSz-yxnmXG3ipU1BBtsEAaQ,2493
 code_loader/experiment_api/cli_config_utils.py,sha256=n6JMyNrquxql3KKxHhAP8jAzezlRT-PV2KWI95kKsm0,1140
@@ -25,8 +25,8 @@ code_loader/leaploader.py,sha256=K__WKfqtKwEch40au177Po10EUX7gm0PJzcV6kpUMlo,222
 code_loader/leaploaderbase.py,sha256=aHlqWDZRacIdBefeB9goYVnpApaNN2FT24uPIWKkCeQ,3090
 code_loader/utils.py,sha256=aw2i_fqW_ADjLB66FWZd9DfpCQ7mPdMyauROC5Nd51I,2197
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-code_loader/visualizers/default_visualizers.py,sha256=VoqO9FN84yXyMjRjHjUTOt2GdTkJRMbHbXJ1cJkREkk,2230
-code_loader-1.0.64.dev3.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.64.dev3.dist-info/METADATA,sha256=ICTjVXWwKwts78r-aY6VX0RzsYxfUpkOTaVeurAc8sk,854
-code_loader-1.0.64.dev3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-code_loader-1.0.64.dev3.dist-info/RECORD,,
+code_loader/visualizers/default_visualizers.py,sha256=Ffx5VHVOe5ujBOsjBSxN_aIEVwFSQ6gbhTMG5aUS-po,2305
+code_loader-1.0.65.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.65.dist-info/METADATA,sha256=hWg-H6gagFGaIzqiHVFmxRXsk568WHJLKzby9YHWlYc,849
+code_loader-1.0.65.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+code_loader-1.0.65.dist-info/RECORD,,