code-loader 1.0.38.dev4__py3-none-any.whl → 1.0.38.dev6__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.
Files changed (6) hide show
  1. code_loader/inner_leap_binder/leapbinder.py +2 -2
  2. code_loader/visualizers/default_visualizers.py +5 -5
  3. {code_loader-1.0.38.dev4.dist-info → code_loader-1.0.38.dev6.dist-info}/METADATA +1 -1
  4. {code_loader-1.0.38.dev4.dist-info → code_loader-1.0.38.dev6.dist-info}/RECORD +6 -6
  5. {code_loader-1.0.38.dev4.dist-info → code_loader-1.0.38.dev6.dist-info}/LICENSE +0 -0
  6. {code_loader-1.0.38.dev4.dist-info → code_loader-1.0.38.dev6.dist-info}/WHEEL +0 -0
code_loader/inner_leap_binder/leapbinder.py CHANGED
@@ -17,7 +17,7 @@ from code_loader.utils import to_numpy_return_wrapper, get_shape
17
17
  from code_loader.visualizers.default_visualizers import DefaultVisualizer, \
18
18
  default_graph_visualizer, \
19
19
  default_image_visualizer, default_horizontal_bar_visualizer, default_word_visualizer, \
20
- default_image_mask_visualizer, default_text_mask_visualizer, default_raw_data_visualizer
20
+ default_image_mask_visualizer, default_text_mask_visualizer, default_raw_data_visualizer, default_image_heatmap_visualizer
21
21
 
22
22
 
23
23
  class LeapBinder:
@@ -31,7 +31,7 @@ class LeapBinder:
31
31
  def _extend_with_default_visualizers(self) -> None:
32
32
  self.set_visualizer(function=default_image_visualizer, name=DefaultVisualizer.Image.value,
33
33
  visualizer_type=LeapDataType.Image)
34
- self.set_visualizer(function=default_image_visualizer, name=DefaultVisualizer.Image.value,
34
+ self.set_visualizer(function=default_image_heatmap_visualizer, name=DefaultVisualizer.ImageHeatmap.value,
35
35
  visualizer_type=LeapDataType.ImageWithHeatmap)
36
36
  self.set_visualizer(function=default_graph_visualizer, name=DefaultVisualizer.Graph.value,
37
37
  visualizer_type=LeapDataType.Graph)
code_loader/visualizers/default_visualizers.py CHANGED
@@ -19,8 +19,12 @@ class DefaultVisualizer(Enum):
19
19
  ImageHeatmap = 'ImageHeatmap'
20
20
 
21
21
 
22
+ def default_image_visualizer(data: npt.NDArray[np.float32]) -> LeapImage:
23
+ rescaled_data = rescale_min_max(data)
24
+ return LeapImage(rescaled_data)
25
+
22
26
 
23
- def default_image_visualizer(image: npt.NDArray[np.float32], heatmaps: npt.NDArray[np.float32], labels: list) -> LeapImageWithHeatmap:
27
+ def default_image_heatmap_visualizer(image: npt.NDArray[np.float32], heatmaps: npt.NDArray[np.float32], labels: list) -> LeapImageWithHeatmap:
24
28
  rescaled_image = rescale_min_max(image)
25
29
  rescaled_heatmaps = []
26
30
  for heatmap in heatmaps:
@@ -28,10 +32,6 @@ def default_image_visualizer(image: npt.NDArray[np.float32], heatmaps: npt.NDArr
28
32
  rescaled_heatmaps = np.append(rescaled_heatmaps, rescaled_heatmap)
29
33
  return LeapImageWithHeatmap(rescaled_image, rescaled_heatmaps, labels)
30
34
 
31
- def default_image_heatmap_visualizer(data: npt.NDArray[np.float32]) -> LeapImage:
32
- rescaled_data = rescale_min_max(data)
33
- return LeapImage(rescaled_data)
34
-
35
35
 
36
36
  def default_graph_visualizer(data: npt.NDArray[np.float32]) -> LeapGraph:
37
37
  return LeapGraph(data)
{code_loader-1.0.38.dev4.dist-info → code_loader-1.0.38.dev6.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: code-loader
3
- Version: 1.0.38.dev4
3
+ Version: 1.0.38.dev6
4
4
  Summary:
5
5
  Home-page: https://github.com/tensorleap/code-loader
6
6
  License: MIT
{code_loader-1.0.38.dev4.dist-info → code_loader-1.0.38.dev6.dist-info}/RECORD RENAMED
@@ -7,12 +7,12 @@ code_loader/contract/exceptions.py,sha256=jWqu5i7t-0IG0jGRsKF4DjJdrsdpJjIYpUkN1F
7
7
  code_loader/contract/responsedataclasses.py,sha256=WSHmFZWOFhGL1eED1u-aoRotPQg2owFQ-t3xSViWXSI,2808
8
8
  code_loader/contract/visualizer_classes.py,sha256=fP7lzIq1SnvxZhfUpC4Q86yxUTaEllYu4pRJ5HaaJdU,5845
9
9
  code_loader/inner_leap_binder/__init__.py,sha256=koOlJyMNYzGbEsoIbXathSmQ-L38N_pEXH_HvL7beXU,99
10
- code_loader/inner_leap_binder/leapbinder.py,sha256=YXXaG89dS1yTLUwGvnEAGvjVMDrNAqVvr53IIiEH77c,13784
10
+ code_loader/inner_leap_binder/leapbinder.py,sha256=KpTAkzP9JCSW6fIWKT1ZRtbmSLcC629UoSWSpt88uQs,13833
11
11
  code_loader/leaploader.py,sha256=pUySweZetJ6SsubCcZlDCJpvWmUrm5YlPlkZWQxY1hQ,17289
12
12
  code_loader/utils.py,sha256=61I4PgSl-ZBIe4DifLxMNlBELE-HQR2pB9efVYPceIU,2230
13
13
  code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
14
- code_loader/visualizers/default_visualizers.py,sha256=Kc-XA59eAb0slArj_KnTb6f5XkVbMsWB2QM0J4k_D4A,2733
15
- code_loader-1.0.38.dev4.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
16
- code_loader-1.0.38.dev4.dist-info/METADATA,sha256=tDsHr7kmSgyYVA5478Tk9hhxS41AjGWWOKK5DonTz7k,773
17
- code_loader-1.0.38.dev4.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
18
- code_loader-1.0.38.dev4.dist-info/RECORD,,
14
+ code_loader/visualizers/default_visualizers.py,sha256=mts54M0JuMkNvBklv_PVqobmj1gcEEO44LxiFAvRivI,2733
15
+ code_loader-1.0.38.dev6.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
16
+ code_loader-1.0.38.dev6.dist-info/METADATA,sha256=x9KacwbSL-1MSrc8n6cXSFUTyeeY5r_fJzOmaoNNawI,773
17
+ code_loader-1.0.38.dev6.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
18
+ code_loader-1.0.38.dev6.dist-info/RECORD,,