code-loader 0.2.86.dev4__py3-none-any.whl → 0.2.88__py3-none-any.whl

code_loader/contract/datasetclasses.py

@@ -43,9 +43,6 @@ class UnlabeledDataPreprocessHandler:
     data_length: int = 0
 
 
-VisualizerCallableReturnType = Union[LeapImage, LeapText, LeapGraph, LeapHorizontalBar,
-                                     LeapImageMask, LeapTextMask, LeapImageWithBBox]
-
 VisualizerCallableInterface = Union[
     Callable[..., LeapImage],
     Callable[..., LeapText],
@@ -53,10 +50,12 @@ VisualizerCallableInterface = Union[
     Callable[..., LeapHorizontalBar],
     Callable[..., LeapImageMask],
     Callable[..., LeapTextMask],
-    Callable[..., LeapImageWithBBox],
-    Callable[..., Dict[str, VisualizerCallableReturnType]]
+    Callable[..., LeapImageWithBBox]
 ]
 
+VisualizerCallableReturnType = Union[LeapImage, LeapText, LeapGraph, LeapHorizontalBar,
+                                     LeapImageMask, LeapTextMask, LeapImageWithBBox]
+
 CustomCallableInterface = Callable[..., tf.Tensor]
 
 

code_loader/leap_binder/leapbinder.py

@@ -10,9 +10,9 @@ from code_loader.contract.datasetclasses import SectionCallableInterface, InputH
     GroundTruthHandler, MetadataHandler, DatasetIntegrationSetup, VisualizerHandler, PreprocessResponse, \
     PreprocessHandler, VisualizerCallableInterface, CustomLossHandler, CustomCallableInterface, PredictionTypeHandler, \
     MetadataSectionCallableInterface, UnlabeledDataPreprocessHandler, CustomLayerHandler, MetricHandler, \
-    ConfusionMatrixCallableInterface, CustomCallableInterfaceMultiArgs, ConfusionMatrixCallableInterfaceMultiArgs, \
+    CustomCallableInterfaceMultiArgs, ConfusionMatrixCallableInterfaceMultiArgs, \
     CustomMultipleReturnCallableInterfaceMultiArgs
-from code_loader.contract.enums import DatasetMetadataType, LeapDataType, MetricEnum
+from code_loader.contract.enums import LeapDataType
 from code_loader.metrics.default_metrics import metrics_names_to_functions
 from code_loader.utils import to_numpy_return_wrapper
 from code_loader.visualizers.default_visualizers import DefaultVisualizer, \
@@ -95,12 +95,7 @@ class LeapBinder:
         self.setup_container.metrics.append(MetricHandler(name, function, arg_names))
 
     @typechecked
-    def add_prediction(self, name: str, labels: List[str], metrics: Optional[List[MetricEnum]] = None,
-                       custom_metrics: Optional[
-                           List[Union[CustomCallableInterface, ConfusionMatrixCallableInterface]]] = None) -> None:
-        if metrics or custom_metrics:
-            raise DeprecationWarning("Adding metrics on 'leap_binder.add_prediction' method is deprecated."
-                                     "Please update the leap script and use metric block instead.")
+    def add_prediction(self, name: str, labels: List[str]) -> None:
         self.setup_container.prediction_types.append(PredictionTypeHandler(name, labels))
 
     @typechecked

code_loader/leaploader.py

@@ -44,6 +44,12 @@ class LeapLoader:
                 torch.cuda.is_available = lambda: False
 
             self.evaluate_module()
+        except TypeError as e:
+            import traceback
+            if "leap_binder.set_metadata(" in traceback.format_exc(5):
+                raise DeprecationWarning(
+                    "Please remove the metadata_type on leap_binder.set_metadata in your dataset script")
+            raise DatasetScriptException(getattr(e, 'message', repr(e))) from e
         except Exception as e:
             raise DatasetScriptException(getattr(e, 'message', repr(e))) from e
 
@@ -280,8 +286,11 @@ class LeapLoader:
                 dataset_metadata_type = DatasetMetadataType.string
             elif metadata_type == bool:
                 dataset_metadata_type = DatasetMetadataType.boolean
+            elif metadata_type == float:
+                dataset_metadata_type = DatasetMetadataType.float
             else:
-                dataset_metadata_type = DatasetMetadataType[metadata_type.__name__]
+                raise Exception(f"Unsupported return type of metadata {handler_test_payload.name}."
+                                f"The return type should be one of [int, float, str, bool]. Got {metadata_type}")
             metadata_instances.append(DatasetMetadataInstance(name=handler_test_payload.name,
                                                               type=dataset_metadata_type))
 

{code_loader-0.2.86.dev4.dist-info → code_loader-0.2.88.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: code-loader
-Version: 0.2.86.dev4
+Version: 0.2.88
 Summary: 
 Home-page: https://github.com/tensorleap/code-loader
 License: MIT
@@ -16,7 +16,7 @@ Requires-Dist: numpy (>=1.22.3,<2.0.0)
 Requires-Dist: psutil (>=5.9.5,<6.0.0)
 Requires-Dist: tensorflow (>=2.11.0,<3.0.0) ; platform_machine == "x86_64"
 Requires-Dist: tensorflow-macos (>=2.11.0,<3.0.0) ; platform_machine == "arm64"
-Requires-Dist: torch (==1.12.1)
+Requires-Dist: torch (>=1.12,<3.0)
 Requires-Dist: typeguard (>=2.13.3,<3.0.0)
 Project-URL: Repository, https://github.com/tensorleap/code-loader
 Description-Content-Type: text/markdown

{code_loader-0.2.86.dev4.dist-info → code_loader-0.2.88.dist-info}/RECORD

@@ -1,7 +1,7 @@
 LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
 code_loader/__init__.py,sha256=ncXAeQmuP80Ml9wgSfusgLn_i6So8j1Hj7e3v4qqDdo,116
 code_loader/contract/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-code_loader/contract/datasetclasses.py,sha256=gOvF-vpmK19s7rMcAovqLCkE_AbcOz9mVyMote3re-0,4614
+code_loader/contract/datasetclasses.py,sha256=TlhPt5HC-Esbhrhe1cmvCSgnE7fnVOXJ-lkzOAspKZ0,4554
 code_loader/contract/enums.py,sha256=KDIeNl79e8rCp4ybJat7U03j50McrcBTWUxko5SXrug,1481
 code_loader/contract/exceptions.py,sha256=jWqu5i7t-0IG0jGRsKF4DjJdrsdpJjIYpUkN1F4RiyQ,51
 code_loader/contract/responsedataclasses.py,sha256=StFpWC5s2lBECZJrpespZ4Ic2iFiSVNgEsxjIHgwgSo,2623
@@ -19,9 +19,9 @@ code_loader/helpers/detection/yolo/utils.py,sha256=BAZmsMz5LrbMlJe5e2SSmrX25E4_C
 code_loader/helpers/instancesegmentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/helpers/instancesegmentation/utils.py,sha256=bl4nlg7mV8CZyPl8grGAE5603TFVM7PTZyM3BRMbXBg,930
 code_loader/leap_binder/__init__.py,sha256=Oe7DNj6RmgX-tMruOrI7GYvf3SyAnzOxugF2e117_Z8,93
-code_loader/leap_binder/leapbinder.py,sha256=BEY33fwG-j2x8HHJj8Lph9fzhUusGwEV2aXQQ51_JNM,6806
+code_loader/leap_binder/leapbinder.py,sha256=-Ny2YewT1jg36Q5AXK8F2Fmv34zxhbcXZp807jIcHws,6293
 code_loader/leap_loader_parallelized_base.py,sha256=_HNbjRK42wRM1V252ETVzibHW8bofFK8-Kq-It5fRks,4758
-code_loader/leaploader.py,sha256=goE3SoOexZ5B78BM0B7PLwfTuEvDNAPzVF6bt-oZFEA,17900
+code_loader/leaploader.py,sha256=iHz5Hq7phFmjxsH1wr6d_Fajwg2JFEBwYzDoXdR46qo,18496
 code_loader/metric_calculator_parallelized.py,sha256=X1FHFlfPorjFVNbms19s_yItkKFja0b97wUJHNvvlxs,2233
 code_loader/metrics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/metrics/default_metrics.py,sha256=xtvTskR7LC5pyKSlFcvlyBpFY9TSXNGVAi5taEPSI60,6727
@@ -29,7 +29,7 @@ code_loader/samples_generator_parallelized.py,sha256=_t0lmf11y1Tas4bdndOSk2b2QMj
 code_loader/utils.py,sha256=WUcM97OuxrhfLCRPoH9EbXrxajNpYgX1CTMc3_PXtYU,1736
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=HqWx2qfTrroGl2n8Fpmr_4X-rk7tE2oGapjO3gzz4WY,2226
-code_loader-0.2.86.dev4.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-0.2.86.dev4.dist-info/METADATA,sha256=qRwT8e8w2QSCXoLuVrkTRdmHb5mXdGvsj4QY_9VaU68,952
-code_loader-0.2.86.dev4.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-code_loader-0.2.86.dev4.dist-info/RECORD,,
+code_loader-0.2.88.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-0.2.88.dist-info/METADATA,sha256=RVdQb6WajrbescPNX6lposmhOIa9dTkWLDOuPL0NS10,950
+code_loader-0.2.88.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+code_loader-0.2.88.dist-info/RECORD,,