code-loader 1.0.189__py3-none-any.whl → 1.0.190.dev0__py3-none-any.whl

code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -197,11 +197,11 @@ def validate_output_structure(result, func_name: str, expected_type_name="np.nda
 def batch_warning(result, func_name):
     if len(result.shape) > 0 and result.shape[0] == 1:
         warnings.warn(
-            f"{func_name} warning: Tensorleap will add a batch dimension at axis 0 to the output of {func_name}, "
-            f"although the detected size of axis 0 is already 1. "
-            f"This may lead to an extra batch dimension (e.g., shape (1, 1, ...)). "
-            f"Please ensure that the output of '{func_name}' is not already batched "
-            f"to avoid computation errors."
+            f"{func_name} warning: Tensorleap will add a batch dimension at axis 0 of "
+            f"{func_name}'s output, but axis 0 is already size 1 — this can produce a "
+            f"doubly-batched shape (e.g. (1, 1, ...)). If axis 0 is a real dimension "
+            f"(e.g. a single channel) and not an accidental batch, you can ignore this; "
+            f"only fix it if you pre-batched the output."
         )
 
 
@@ -349,6 +349,14 @@ def _safe_get_item(key):
 
 
 def tensorleap_load_model(prediction_types: Optional[List[PredictionTypeHandler]] = _UNSET):
+    """Register the model-loading function.
+
+    ``prediction_types``: declare one ``PredictionTypeHandler`` per model output, in
+    output order. Multi-output models must declare an entry for EVERY output — the
+    count is validated against the model's outputs — so for an output you don't consume
+    as a prediction, pass a throwaway handler to satisfy the count. Omit the argument
+    only when the model has a single prediction output.
+    """
     prediction_types_was_provided = prediction_types is not _UNSET
 
     if not prediction_types_was_provided:
@@ -567,6 +575,9 @@ def tensorleap_load_model(prediction_types: Optional[List[PredictionTypeHandler]
                 def get_inputs(self):
                     return self.model.get_inputs()
 
+                def get_outputs(self):
+                    return self.model.get_outputs()
+
             model_placeholder = ModelPlaceholder(prediction_types)
             if not _call_from_tl_platform:
                 update_env_params_func("tensorleap_load_model", "v")
@@ -631,6 +642,13 @@ def tensorleap_load_model(prediction_types: Optional[List[PredictionTypeHandler]
 
                     return FollowInputIndex()
 
+                def get_outputs(self):
+                    # Mapping-mode counterpart of the ONNX-run get_outputs(): in mapping
+                    # mode outputs are reached by index, and get_outputs()[i] resolves to
+                    # Prediction{i} — exactly like run()/__call__, which also return a
+                    # ModelOutputPlaceholder. Reusing it keeps the mapping unchanged.
+                    return ModelOutputPlaceholder()
+
             return ModelPlaceholder()
 
         def final_inner(*args, **kwargs):

code_loader/leaploader.py

@@ -320,9 +320,6 @@ class LeapLoader(LeapLoaderBase):
             try:
                 self.exec_script()
 
-                integration_test_test_payload = self._check_integration_test_exists()
-                test_payloads.append(integration_test_test_payload)
-
                 preprocess_test_payload = self._check_preprocess()
                 test_payloads.append(preprocess_test_payload)
                 handlers_test_payloads = self._check_handlers()
@@ -359,23 +356,6 @@ class LeapLoader(LeapLoaderBase):
                                        engine_file_contract=EngineFileContract(global_leap_binder.mapping_connections,
                                                                                global_leap_binder.leap_analysis_configuration))
 
-    def _check_integration_test_exists(self) -> DatasetTestResultPayload:
-        test_result = DatasetTestResultPayload('integration_test')
-        # Only enforced on the Tensorleap platform (i.e. during a push / inspection,
-        # where IS_TENSORLEAP_PLATFORM=true). Running code-loader locally or in unit
-        # tests must not require the decorator, so pre-decorator integrations and
-        # local check runs keep working unchanged.
-        is_on_platform = os.environ.get('IS_TENSORLEAP_PLATFORM') == 'true'
-        if is_on_platform and global_leap_binder.integration_test_func is None:
-            test_result.is_passed = False
-            test_result.display[TestingSectionEnum.Errors.name] = (
-                "No integration test was found in your integration file. A valid Tensorleap "
-                "integration file must define an integration test function decorated with "
-                "@tensorleap_integration_test. Without it the integration cannot be validated "
-                "and the push is aborted. See the mnist leap_integration.py for a valid example."
-            )
-        return test_result
-
     def _check_preprocess(self) -> DatasetTestResultPayload:
         test_result = DatasetTestResultPayload('preprocess')
         try:

{code_loader-1.0.189.dist-info → code_loader-1.0.190.dev0.dist-info}/METADATA

@@ -1,8 +1,7 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: code-loader
-Version: 1.0.189
+Version: 1.0.190.dev0
 Summary: 
-Home-page: https://github.com/tensorleap/code-loader
 License: MIT
 Author: dorhar
 Author-email: doron.harnoy@tensorleap.ai
@@ -20,6 +19,7 @@ Requires-Dist: numpy (>=2.3.2,<3.0.0) ; python_version >= "3.11" and python_vers
 Requires-Dist: psutil (>=5.9.5,<6.0.0)
 Requires-Dist: pyyaml (>=6.0.2,<7.0.0)
 Requires-Dist: requests (>=2.32.3,<3.0.0)
+Project-URL: Homepage, https://github.com/tensorleap/code-loader
 Project-URL: Repository, https://github.com/tensorleap/code-loader
 Description-Content-Type: text/markdown
 

{code_loader-1.0.189.dist-info → code_loader-1.0.190.dev0.dist-info}/RECORD

@@ -1,4 +1,3 @@
-LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
 code_loader/__init__.py,sha256=outxRQ0M-zMfV0QGVJmAed5qWfRmyD0TV6-goEGAzBw,406
 code_loader/contract/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/contract/datasetclasses.py,sha256=RYZcnX7vLMYhyQtCbX2PscVFNAYULxgAA9U4DRRfLqA,10456
@@ -22,8 +21,8 @@ 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=PAX9cmT1QQV7PWsbY9rMRWktpM7KryfFlwBiSwSskzg,42654
-code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=k7_kJaEHEuoKWAGRUDXqlZmR1C228xA_Ny9mmliYKys,122098
-code_loader/leaploader.py,sha256=m2niP0L7liNwxGJHkcxXv28NCRY8MJ2aE3xVVzx4xuM,45969
+code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=lgKJ9cJDuDABscDqLvnqRt7ViU6YeZFcm9996EUosGA,123107
+code_loader/leaploader.py,sha256=K9Q5kiCZ_A2GkS5qOEantAMvWGaz3bmO1X8buXDSpgg,44682
 code_loader/leaploaderbase.py,sha256=l36qDA00GhZEG5NLKpEtAXgWJA-UQQIhNFGxywK7mUA,6530
 code_loader/mixpanel_tracker.py,sha256=rNwRmFifNbdUoqLQvvhhgpKczWpWiEmd8MfyJe27sxw,9131
 code_loader/plot_functions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -32,7 +31,7 @@ code_loader/plot_functions/visualize.py,sha256=gsBAYYkwMh7jIpJeDMPS8G4CW-pxwx6Lz
 code_loader/utils.py,sha256=YecipkdTA-VcE9F0RQcY9cFnY8P3AksPnHM2Db7xUSk,3972
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=onRnLE_TXfgLN4o52hQIOOhUcFexGlqJ3xSpQDVLuZM,2604
-code_loader-1.0.189.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.189.dist-info/METADATA,sha256=Lp2cTL4qNF6QGpuKHvSTtKdAhQgynGI-9gFshYEWA3U,1090
-code_loader-1.0.189.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-code_loader-1.0.189.dist-info/RECORD,,
+code_loader-1.0.190.dev0.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.190.dev0.dist-info/METADATA,sha256=2Tqb-fK_hBKdjefZN31fYytgQdg3VS9rgcPiwqLsLkc,1107
+code_loader-1.0.190.dev0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+code_loader-1.0.190.dev0.dist-info/RECORD,,

{code_loader-1.0.189.dist-info → code_loader-1.0.190.dev0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.9.0
+Generator: poetry-core 2.1.3
 Root-Is-Purelib: true
 Tag: py3-none-any

code_loader-1.0.189.dist-info/LICENSE

@@ -1,21 +0,0 @@
-MIT License
-
-Copyright (c) 2021 TensorLeap
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.