code-loader 1.0.193.dev3__py3-none-any.whl → 1.0.193.dev4__py3-none-any.whl

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.dist-info → code_loader-1.0.193.dev4.dist-info}/METADATA

@@ -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.dist-info → code_loader-1.0.193.dev4.dist-info}/RECORD

@@ -22,7 +22,7 @@ code_loader/experiment_api/utils.py,sha256=XZHtxge12TS4H4-8PjV3sKuhp8Ud6ojAiIzTZ
 code_loader/experiment_api/workingspace_config_utils.py,sha256=DLzXQCg4dgTV_YgaSbeTVzq-2ja_SQw4zi7LXwKL9cY,990
 code_loader/inner_leap_binder/__init__.py,sha256=koOlJyMNYzGbEsoIbXathSmQ-L38N_pEXH_HvL7beXU,99
 code_loader/inner_leap_binder/leapbinder.py,sha256=4cbSEbN6vVwjFUMxDEksxk1VSCip2ntLB32RxA81_JE,53194
-code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=EMg6o1wDl3HMFm8n6kEjyprXUX1UKgcOZjNJQCBSu_I,174865
+code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=ShgqKiyfcJKMIpoB_QdxHoQJ0S1gdQDEQgoPIv6sSIw,176492
 code_loader/leaploader.py,sha256=sx-BzeVG3qt54S2uF5VbBi2xL0x70AIFN1to6OtjBYU,66979
 code_loader/leaploaderbase.py,sha256=ncKEWsVv_1Vhtgl8Dqgt4ampxTzOj5nVitGJwQQtTc4,10561
 code_loader/mixpanel_tracker.py,sha256=rNwRmFifNbdUoqLQvvhhgpKczWpWiEmd8MfyJe27sxw,9131
@@ -32,7 +32,7 @@ code_loader/plot_functions/visualize.py,sha256=gsBAYYkwMh7jIpJeDMPS8G4CW-pxwx6Lz
 code_loader/utils.py,sha256=1XLY5hNQaOsqXeUIooGs6KXV0E1edeiQyxAdnG3eiy0,6338
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=onRnLE_TXfgLN4o52hQIOOhUcFexGlqJ3xSpQDVLuZM,2604
-code_loader-1.0.193.dev3.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.193.dev3.dist-info/METADATA,sha256=3rGLwq82vR91AHkFJXYOaCCe_KRiwDBVWlkZYqWMKk4,1095
-code_loader-1.0.193.dev3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-code_loader-1.0.193.dev3.dist-info/RECORD,,
+code_loader-1.0.193.dev4.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.193.dev4.dist-info/METADATA,sha256=0HFcGksXyXxl3m20DjXWkMzAxAXZ7Zt3O0Vl8a8h0K4,1095
+code_loader-1.0.193.dev4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+code_loader-1.0.193.dev4.dist-info/RECORD,,