code-loader 1.0.184.dev0__tar.gz → 1.0.184.dev2__tar.gz

{code_loader-1.0.184.dev0 → code_loader-1.0.184.dev2}/PKG-INFO

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

{code_loader-1.0.184.dev0 → code_loader-1.0.184.dev2}/code_loader/leaploader.py

@@ -452,12 +452,14 @@ class LeapLoader(LeapLoaderBase):
                 data={},
                 state=DataStateType.additional,
                 sample_id_type=str,
+                tl_generated=True,
             )
-            placeholder.tl_generated = True
             self._preprocess_result_cached[DataStateEnum.additional] = placeholder
+            global_leap_binder.setup_container.preprocess.data_length[DataStateType.additional] = placeholder.length
         else:
             additional.sample_ids = list(additional.sample_ids) + list(new_sample_ids)
             additional.length = len(additional.sample_ids)
+            global_leap_binder.setup_container.preprocess.data_length[DataStateType.additional] = additional.length
 
     @staticmethod
     def _get_all_dataset_base_handlers() -> List[Union[DatasetBaseHandler, MetadataHandler]]:
@@ -805,17 +807,9 @@ class LeapLoader(LeapLoaderBase):
         return global_leap_binder.setup_container.custom_latent_space is not None
 
     def get_instances_data(self, state: DataStateEnum) -> Tuple[Dict[str, List[str]], Dict[str, str]]:
-        """
-        This Method get the data state and returns two dictionaries that holds the mapping of the sample ids to their
-         instances and the other way around and the sample ids array.
-        Args:
-            state: DataStateEnum state
-        Returns:
-            sample_ids_to_instance_mappings: sample id to instance mappings
-            instance_to_sample_ids_mappings: instance to sample ids mappings
-            sample_ids: sample ids array
-        """
         preprocess_result = self._preprocess_result()
-        preprocess_state = preprocess_result[state]
+        preprocess_state = preprocess_result.get(state)
+        if preprocess_state is None:
+            return None, None
         return preprocess_state.sample_ids_to_instance_mappings, preprocess_state.instance_to_sample_ids_mappings
 

{code_loader-1.0.184.dev0 → code_loader-1.0.184.dev2}/pyproject.toml

@@ -1,7 +1,7 @@
 [tool.poetry]
 name = "code-loader"
 
-version = "1.0.184.dev0"
+version = "1.0.184.dev2"
 description = ""
 authors = ["dorhar <doron.harnoy@tensorleap.ai>"]
 license = "MIT"