code-loader 1.0.206.dev2__tar.gz → 1.0.206.dev3__tar.gz

{code_loader-1.0.206.dev2 → code_loader-1.0.206.dev3}/PKG-INFO

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

{code_loader-1.0.206.dev2 → code_loader-1.0.206.dev3}/code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -3017,6 +3017,11 @@ def tensorleap_element_instance_preprocess(
             resolved = {}
             for name, value in element_instance_metadata.items():
                 if name in declared:
+                    # Still type-check the VALUE. Master ran every key through
+                    # map_dict_to_metadata_types, so declaring a key must not become a way to
+                    # smuggle a list or an ndarray past validation and have it fail at eval time.
+                    if not is_absent_metadata_value(value):
+                        get_metadata_type_from_variable(value)
                     continue
                 if is_absent_metadata_value(value):
                     raise Exception(
@@ -3034,11 +3039,18 @@ def tensorleap_element_instance_preprocess(
             found_instance_metadata = False
             # Resolve types on training, like mechanism (a) does (leapbinder.get_preprocess_result):
             # only training is allowed to fail on an unresolvable absence. `state` is still None here
-            # whenever the integration did not set it, so fall back to list position -- which is what
-            # get_preprocess_result would assign, and position 0 is training.
-            probe_order = sorted(
-                range(len(result)),
-                key=lambda i: DataStateEnum[result[i].state.name] if result[i].state is not None else i)
+            # whenever the integration did not set it, so fall back to list position -- which is the
+            # state get_preprocess_result would assign, and position 0 is training. Rank training
+            # against everything else rather than sorting on the state value itself: mixing enum
+            # values with list positions in one key lets a validation response tie with position 1
+            # and keep its place ahead of training.
+            def _is_training(i: int) -> bool:
+                response = result[i]
+                if response.state is not None:
+                    return response.state == DataStateType.training
+                return i == 0
+
+            probe_order = sorted(range(len(result)), key=lambda i: (0 if _is_training(i) else 1, i))
             for preprocess_response in (result[i] for i in probe_order):
                 if preprocess_response.is_grouped:
                     raise Exception(

{code_loader-1.0.206.dev2 → code_loader-1.0.206.dev3}/pyproject.toml

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