code-loader 1.0.64.dev2__py3-none-any.whl → 1.0.64.dev4__py3-none-any.whl

code_loader/leaploader.py

@@ -245,11 +245,12 @@ class LeapLoader(LeapLoaderBase):
         return self._visualizer_handler_by_name()[visualizer_name].function(**input_tensors_by_arg_name)
 
     def run_heatmap_visualizer(self, visualizer_name: str, input_tensors_by_arg_name: Dict[str, npt.NDArray[np.float32]]
-                               ) -> npt.NDArray[np.float32]:
+                               ) -> Optional[npt.NDArray[np.float32]]:
         heatmap_function = self._visualizer_handler_by_name()[visualizer_name].heatmap_function
         if heatmap_function is None:
             assert len(input_tensors_by_arg_name) == 1
-            return list(input_tensors_by_arg_name.values())[0]
+            return None
+
         return heatmap_function(**input_tensors_by_arg_name)
 
     def get_heatmap_visualizer_raw_vis_input_arg_name(self, visualizer_name: str) -> Optional[str]:

code_loader/leaploaderbase.py

@@ -64,7 +64,7 @@ class LeapLoaderBase:
 
     @abstractmethod
     def run_heatmap_visualizer(self, visualizer_name: str, input_tensors_by_arg_name: Dict[str, npt.NDArray[np.float32]]
-                               ) -> npt.NDArray[np.float32]:
+                               ) -> Optional[npt.NDArray[np.float32]]:
         pass
 
     @abstractmethod

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.dev2.dist-info → code_loader-1.0.64.dev4.dist-info}/METADATA

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

{code_loader-1.0.64.dev2.dist-info → code_loader-1.0.64.dev4.dist-info}/RECORD

@@ -21,12 +21,12 @@ code_loader/experiment_api/workingspace_config_utils.py,sha256=DLzXQCg4dgTV_YgaS
 code_loader/inner_leap_binder/__init__.py,sha256=koOlJyMNYzGbEsoIbXathSmQ-L38N_pEXH_HvL7beXU,99
 code_loader/inner_leap_binder/leapbinder.py,sha256=y_k7bRFYYmrZo4jCJrZ6mJykxc1slKDkODYCZ58OPs0,25691
 code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=I6ipji6QMN9qqFYxqQyNjtDtsnwpB-NahgKKpLPctMo,21067
-code_loader/leaploader.py,sha256=vssXiifNEGkfapjDtVd0Lpg3WKQ6qxYyCdoeGWvxiNc,22240
-code_loader/leaploaderbase.py,sha256=I1xwgGnDlvfyoOhlpKXXEuCqJUs0jqPzk3DTQaE6riQ,3080
+code_loader/leaploader.py,sha256=K__WKfqtKwEch40au177Po10EUX7gm0PJzcV6kpUMlo,22212
+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.dev2.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.64.dev2.dist-info/METADATA,sha256=vKLtUJvnmvuG4wZ6pkkM54P6Z54v1IOd_PnGwYDJl-s,854
-code_loader-1.0.64.dev2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-code_loader-1.0.64.dev2.dist-info/RECORD,,
+code_loader/visualizers/default_visualizers.py,sha256=Ffx5VHVOe5ujBOsjBSxN_aIEVwFSQ6gbhTMG5aUS-po,2305
+code_loader-1.0.64.dev4.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.64.dev4.dist-info/METADATA,sha256=aOuZry-bit9EU0-lQKuygX4p-Usgx1mWs4olOy4w0M8,854
+code_loader-1.0.64.dev4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+code_loader-1.0.64.dev4.dist-info/RECORD,,