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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
code_loader/leaploader.py CHANGED
@@ -245,11 +245,12 @@ class LeapLoader(LeapLoaderBase):
245
245
  return self._visualizer_handler_by_name()[visualizer_name].function(**input_tensors_by_arg_name)
246
246
 
247
247
  def run_heatmap_visualizer(self, visualizer_name: str, input_tensors_by_arg_name: Dict[str, npt.NDArray[np.float32]]
248
- ) -> npt.NDArray[np.float32]:
248
+ ) -> Optional[npt.NDArray[np.float32]]:
249
249
  heatmap_function = self._visualizer_handler_by_name()[visualizer_name].heatmap_function
250
250
  if heatmap_function is None:
251
251
  assert len(input_tensors_by_arg_name) == 1
252
- return list(input_tensors_by_arg_name.values())[0]
252
+ return None
253
+
253
254
  return heatmap_function(**input_tensors_by_arg_name)
254
255
 
255
256
  def get_heatmap_visualizer_raw_vis_input_arg_name(self, visualizer_name: str) -> Optional[str]:
@@ -64,7 +64,7 @@ class LeapLoaderBase:
64
64
 
65
65
  @abstractmethod
66
66
  def run_heatmap_visualizer(self, visualizer_name: str, input_tensors_by_arg_name: Dict[str, npt.NDArray[np.float32]]
67
- ) -> npt.NDArray[np.float32]:
67
+ ) -> Optional[npt.NDArray[np.float32]]:
68
68
  pass
69
69
 
70
70
  @abstractmethod
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: code-loader
3
- Version: 1.0.64.dev2
3
+ Version: 1.0.64.dev3
4
4
  Summary:
5
5
  Home-page: https://github.com/tensorleap/code-loader
6
6
  License: MIT
@@ -21,12 +21,12 @@ code_loader/experiment_api/workingspace_config_utils.py,sha256=DLzXQCg4dgTV_YgaS
21
21
  code_loader/inner_leap_binder/__init__.py,sha256=koOlJyMNYzGbEsoIbXathSmQ-L38N_pEXH_HvL7beXU,99
22
22
  code_loader/inner_leap_binder/leapbinder.py,sha256=y_k7bRFYYmrZo4jCJrZ6mJykxc1slKDkODYCZ58OPs0,25691
23
23
  code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=I6ipji6QMN9qqFYxqQyNjtDtsnwpB-NahgKKpLPctMo,21067
24
- code_loader/leaploader.py,sha256=vssXiifNEGkfapjDtVd0Lpg3WKQ6qxYyCdoeGWvxiNc,22240
25
- code_loader/leaploaderbase.py,sha256=I1xwgGnDlvfyoOhlpKXXEuCqJUs0jqPzk3DTQaE6riQ,3080
24
+ code_loader/leaploader.py,sha256=K__WKfqtKwEch40au177Po10EUX7gm0PJzcV6kpUMlo,22212
25
+ code_loader/leaploaderbase.py,sha256=aHlqWDZRacIdBefeB9goYVnpApaNN2FT24uPIWKkCeQ,3090
26
26
  code_loader/utils.py,sha256=aw2i_fqW_ADjLB66FWZd9DfpCQ7mPdMyauROC5Nd51I,2197
27
27
  code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
28
28
  code_loader/visualizers/default_visualizers.py,sha256=VoqO9FN84yXyMjRjHjUTOt2GdTkJRMbHbXJ1cJkREkk,2230
29
- code_loader-1.0.64.dev2.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
30
- code_loader-1.0.64.dev2.dist-info/METADATA,sha256=vKLtUJvnmvuG4wZ6pkkM54P6Z54v1IOd_PnGwYDJl-s,854
31
- code_loader-1.0.64.dev2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
32
- code_loader-1.0.64.dev2.dist-info/RECORD,,
29
+ code_loader-1.0.64.dev3.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
30
+ code_loader-1.0.64.dev3.dist-info/METADATA,sha256=ICTjVXWwKwts78r-aY6VX0RzsYxfUpkOTaVeurAc8sk,854
31
+ code_loader-1.0.64.dev3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
32
+ code_loader-1.0.64.dev3.dist-info/RECORD,,