code-loader 1.0.25__py3-none-any.whl → 1.0.26__py3-none-any.whl

code_loader/leaploader.py

@@ -5,7 +5,7 @@ import sys
 from contextlib import redirect_stdout
 from functools import lru_cache
 from pathlib import Path
-from typing import Dict, List, Iterable, Union
+from typing import Dict, List, Iterable, Union, Any
 
 import numpy as np
 import numpy.typing as npt
@@ -301,7 +301,40 @@ class LeapLoader:
     def _get_gt(self, state: DataStateEnum, idx: int) -> Dict[str, npt.NDArray[np.float32]]:
         return self._get_dataset_handlers(global_leap_binder.setup_container.ground_truths, state, idx)
 
+    @lru_cache()
+    def _metadata_name_to_type(self) -> Dict[str, DatasetMetadataType]:
+        global_leap_binder.check_preprocess(self._preprocess_result())
+        handlers_test_payloads = self._check_handlers()
+        metadata_setup = self.get_dataset_setup_response(handlers_test_payloads).metadata
+        metadata_name_to_type = {
+            metadata_instance.name: metadata_instance.type
+            for metadata_instance in metadata_setup
+        }
+        return metadata_name_to_type
+
     def _get_metadata(self, state: DataStateEnum, idx: int) -> Dict[str, Union[str, int, bool, float]]:
+        def _convert_metadata_to_correct_type(metadata_name: str, value: Any) -> Any:
+            metadata_name_to_type = self._metadata_name_to_type()
+            metadata_type_to_python_type = {
+                DatasetMetadataType.float: float,
+                DatasetMetadataType.string: str,
+                DatasetMetadataType.boolean: bool,
+                DatasetMetadataType.int: int
+            }
+            metadata_type_to_default_value = {
+                DatasetMetadataType.float: -1,
+                DatasetMetadataType.string: "",
+                DatasetMetadataType.boolean: False,
+                DatasetMetadataType.int: -1
+            }
+
+            try:
+                converted_value = metadata_type_to_python_type[metadata_name_to_type[metadata_name]](value)
+            except ValueError:
+                converted_value = metadata_type_to_default_value[metadata_name_to_type[metadata_name]]
+
+            return converted_value
+
         result_agg = {}
         preprocess_result = self._preprocess_result()
         preprocess_state = preprocess_result[state]
@@ -310,9 +343,9 @@ class LeapLoader:
             if isinstance(handler_result, dict):
                 for single_metadata_name, single_metadata_result in handler_result.items():
                     handler_name = f'{handler.name}_{single_metadata_name}'
-                    result_agg[handler_name] = single_metadata_result
+                    result_agg[handler_name] = _convert_metadata_to_correct_type(handler_name, single_metadata_result)
             else:
                 handler_name = handler.name
-                result_agg[handler_name] = handler_result
+                result_agg[handler_name] = _convert_metadata_to_correct_type(handler_name, handler_result)
 
         return result_agg

{code_loader-1.0.25.dist-info → code_loader-1.0.26.dist-info}/METADATA

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

{code_loader-1.0.25.dist-info → code_loader-1.0.26.dist-info}/RECORD

@@ -9,11 +9,11 @@ code_loader/contract/responsedataclasses.py,sha256=0bo4iLbfu6-3tpgzeKakIH2i_1jtR
 code_loader/contract/visualizer_classes.py,sha256=1FjVO744J_EMuJfHWXGdvSz6vl3Vu7iS3CDfs8MzEEQ,5138
 code_loader/inner_leap_binder/__init__.py,sha256=koOlJyMNYzGbEsoIbXathSmQ-L38N_pEXH_HvL7beXU,99
 code_loader/inner_leap_binder/leapbinder.py,sha256=Nb3UXrtoYMDNLGbL6JvIIaVMEAQ_B8YpUPXxZ5DThrE,12540
-code_loader/leaploader.py,sha256=g8zpaCSmR-oNbgMt4wvmAJ2fBRut_n9d5WnCcHCw8rA,15590
+code_loader/leaploader.py,sha256=iD4s8fHD1zqTzitc0Gxj7MeZhTFXKf6t77P_glvHOFI,17147
 code_loader/utils.py,sha256=DpzbjnCOWAS-XPJg-tUgqtMUOSps3AMd2FDL589d6Bk,2135
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=HqWx2qfTrroGl2n8Fpmr_4X-rk7tE2oGapjO3gzz4WY,2226
-code_loader-1.0.25.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.25.dist-info/METADATA,sha256=X9PB6W0DZsHiHTvkXqv56wPjVLqE4xE2IbhUJKW4RuY,768
-code_loader-1.0.25.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-code_loader-1.0.25.dist-info/RECORD,,
+code_loader-1.0.26.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.26.dist-info/METADATA,sha256=z39LgR3lLMsLSltb6epLwrWHMJfB996P7kE47QdEImM,768
+code_loader-1.0.26.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+code_loader-1.0.26.dist-info/RECORD,,