code-loader 1.0.193.dev3__tar.gz → 1.0.193.dev4__tar.gz

{code_loader-1.0.193.dev3 → code_loader-1.0.193.dev4}/PKG-INFO

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

{code_loader-1.0.193.dev3 → code_loader-1.0.193.dev4}/code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -299,8 +299,19 @@ def tensorleap_integration_test():
         def inner(*args, **kwargs):
             if not _call_from_tl_platform:
                 set_current('tensorleap_integration_test')
+            # Keyword calls must work with the user function's OWN parameter names — the local
+            # run template calls integration_test(sample_id=..., preprocess=...) against a
+            # signature the user chose, not the canonical placeholder names.
+            expected_names = inspect.getfullargspec(integration_test_function)[0][:2]
+            if len(expected_names) < 2:
+                expected_names = ['idx', 'preprocess']
             validate_args_structure(*args, types_order=[Union[int, str], PreprocessResponse],
-                                    func_name='integration_test', expected_names=["idx", "preprocess"], **kwargs)
+                                    func_name='integration_test', expected_names=expected_names,
+                                    **kwargs)
+            sample_id, preprocess_response = (
+                args[i] if i < len(args) else kwargs[name]
+                for i, name in enumerate(expected_names))
+            args, kwargs = (sample_id, preprocess_response), {}
             _validate_input_args(*args, **kwargs)
 
             global _called_from_inside_tl_integration_test_decorator, _integration_test_started
@@ -433,7 +444,7 @@ def tensorleap_load_model(prediction_types: Optional[List[PredictionTypeHandler]
             pass
 
         @lru_cache()
-        def inner(*args, **kwargs):
+        def _cached_inner(*args, **kwargs):
             if not _call_from_tl_platform:
                 set_current('tensorleap_load_model')
             validate_args_structure(*args, types_order=[],
@@ -587,12 +598,20 @@ def tensorleap_load_model(prediction_types: Optional[List[PredictionTypeHandler]
                     return self.model.get_inputs()
 
             model_placeholder = ModelPlaceholder(prediction_types)
-            global _last_loaded_model
-            _last_loaded_model = model_placeholder
             if not _call_from_tl_platform:
                 update_env_params_func("tensorleap_load_model", "v")
             return model_placeholder
 
+        def inner(*args, **kwargs):
+            model_placeholder = _cached_inner(*args, **kwargs)
+            # Re-assert on every call, outside the cache: each mapping rerun's load_model
+            # overwrites this global with its mapping placeholder, and on a cache hit the
+            # body above never runs — a second integration test would otherwise fail the
+            # model loop's identity check against the stale mapping placeholder.
+            global _last_loaded_model
+            _last_loaded_model = model_placeholder
+            return model_placeholder
+
         def mapping_inner():
             class ModelOutputPlaceholder:
                 def __init__(self):
@@ -3376,6 +3395,12 @@ def tensorleap_status_table():
         if not started_from("leap_integration.py"):
             return
 
+        if leap_binder.setup_container.autoregressive_step is not None:
+            # Input encoders are forbidden in autoregressive integrations — the step hook is
+            # the sole source of model inputs — so the row is not applicable: it must not be
+            # printed, gate readiness, or be recommended as a next step.
+            table[:] = [row for row in table if row["name"] != "tensorleap_input_encoder"]
+
         ready_mess = "\nAll parts have been successfully set. If no errors accured, you can now push the project to the Tensorleap system."
         not_ready_mess = "\nSome mandatory components have not yet been added to the Integration test. Recommended next interface to add is: "
         mandatory_ready_mess = "\nAll mandatory parts have been successfully set. If no errors accured, you can now push the project to the Tensorleap system or continue to the next optional reccomeded interface,adding: "

{code_loader-1.0.193.dev3 → code_loader-1.0.193.dev4}/pyproject.toml

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