code-loader 1.0.189__tar.gz → 1.0.190.dev0__tar.gz

{code_loader-1.0.189 → code_loader-1.0.190.dev0}/PKG-INFO

@@ -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 → code_loader-1.0.190.dev0}/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-1.0.189 → code_loader-1.0.190.dev0}/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 → code_loader-1.0.190.dev0}/pyproject.toml

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