code-loader 1.0.70.dev0__py3-none-any.whl → 1.0.70.dev2__py3-none-any.whl

code_loader/contract/visualizer_classes.py

@@ -34,6 +34,7 @@ class LeapImage:
     Attributes:
     data (npt.NDArray[np.float32] | npt.NDArray[np.uint8]): The image data.
     type (LeapDataType): The data type, default is LeapDataType.Image.
+    compress: Optional[bool]: Whether to compress the image (.jpg) or not (.png)
 
     Example:
         image_data = np.random.rand(100, 100, 3).astype(np.float32)
@@ -41,6 +42,7 @@ class LeapImage:
     """
     data: Union[npt.NDArray[np.float32], npt.NDArray[np.uint8]]
     type: LeapDataType = LeapDataType.Image
+    compress: Optional[bool] = True
 
     def __post_init__(self) -> None:
         validate_type(self.type, LeapDataType.Image)
@@ -48,6 +50,7 @@ class LeapImage:
         validate_type(self.data.dtype, [np.uint8, np.float32])
         validate_type(len(self.data.shape), 3, 'Image must be of shape 3')
         validate_type(self.data.shape[2], [1, 3], 'Image channel must be either 3(rgb) or 1(gray)')
+        validate_type(type(self.compress), bool, 'compress flag must be a boolean')
 
 
 @dataclass
@@ -172,12 +175,18 @@ class LeapHorizontalBar:
     body: npt.NDArray[np.float32]
     labels: List[str]
     type: LeapDataType = LeapDataType.HorizontalBar
+    gt: Optional[npt.NDArray[np.float32]] = None
+
 
     def __post_init__(self) -> None:
         validate_type(self.type, LeapDataType.HorizontalBar)
         validate_type(type(self.body), np.ndarray)
         validate_type(self.body.dtype, np.float32)
         validate_type(len(self.body.shape), 1, 'HorizontalBar body must be of shape 1')
+        validate_type(type(self.gt), [np.ndarray, type(None)],
+                      'Provided HorizontalBar GT data should be a numpy array')
+        if self.gt is not None:
+            validate_type(len(self.gt.shape), 1, 'HorizontalBar GT must be of shape 1')
 
         validate_type(type(self.labels), list)
         for label in self.labels:

{code_loader-1.0.70.dev0.dist-info → code_loader-1.0.70.dev2.dist-info}/METADATA

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

{code_loader-1.0.70.dev0.dist-info → code_loader-1.0.70.dev2.dist-info}/RECORD

@@ -5,7 +5,7 @@ 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=I2VK4szHwTpdKn_yS-hNCAAG59x4Fk8pqYsgyicRdDA,13578
+code_loader/contract/visualizer_classes.py,sha256=WpO5KF4FMGGMgsF1vIs7nbVev_Mr5-VE9XPakDk87zM,14092
 code_loader/default_losses.py,sha256=NoOQym1106bDN5dcIk56Elr7ZG5quUHArqfP5-Nyxyo,1139
 code_loader/default_metrics.py,sha256=v16Mrt2Ze1tXPgfKywGVdRSrkaK4CKLNQztN1UdVqIY,5010
 code_loader/experiment_api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -26,7 +26,7 @@ code_loader/leaploaderbase.py,sha256=aHlqWDZRacIdBefeB9goYVnpApaNN2FT24uPIWKkCeQ
 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=Ffx5VHVOe5ujBOsjBSxN_aIEVwFSQ6gbhTMG5aUS-po,2305
-code_loader-1.0.70.dev0.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.70.dev0.dist-info/METADATA,sha256=dzFXWTxSULeZSNLBrhqFT1pQzLucSgNuBsDUbFfm7go,854
-code_loader-1.0.70.dev0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-code_loader-1.0.70.dev0.dist-info/RECORD,,
+code_loader-1.0.70.dev2.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.70.dev2.dist-info/METADATA,sha256=0LENVQoZ9j3Qhp31Im2VREi6rSsZB7iFaRP5nL5AMZU,854
+code_loader-1.0.70.dev2.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+code_loader-1.0.70.dev2.dist-info/RECORD,,