code-loader 1.0.187__tar.gz → 1.0.188.dev0__tar.gz

{code_loader-1.0.187 → code_loader-1.0.188.dev0}/PKG-INFO

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

{code_loader-1.0.187 → code_loader-1.0.188.dev0}/code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -768,18 +768,14 @@ def tensorleap_custom_metric(name: str,
 
                 leap_binder.setup_container.metrics[-1].metric_handler_data.direction = effective_direction
 
-                if compute_insights is None:
-                    effective_compute_insights = None
-                elif isinstance(compute_insights, dict):
-                    effective_compute_insights = {key: compute_insights.get(key, True) for key in result_keys}
-                else:
-                    effective_compute_insights = {key: compute_insights for key in result_keys}
-
-                leap_binder.setup_container.metrics[-1].metric_handler_data.compute_insights = effective_compute_insights
-
                 if defaulted_direction_keys and not _call_from_tl_platform:
-                    warning_keys = {key for key in defaulted_direction_keys
-                                    if (effective_compute_insights or {}).get(key, True)}
+                    if compute_insights is None:
+                        effective_compute_insights = {}
+                    elif isinstance(compute_insights, dict):
+                        effective_compute_insights = compute_insights
+                    else:
+                        effective_compute_insights = {k: compute_insights for k in result_keys}
+                    warning_keys = {key for key in defaulted_direction_keys if effective_compute_insights.get(key, True)}
                     if warning_keys:
                         store_warning_by_param(
                             param_name=f"direction[{warning_keys}]",

{code_loader-1.0.187 → code_loader-1.0.188.dev0}/code_loader/leaploader.py

@@ -64,6 +64,9 @@ class LeapLoader(LeapLoaderBase):
             self.evaluate_module()
             if global_leap_binder.integration_test_func is not None:
                 global_leap_binder.integration_test_func(None, PreprocessResponse(state=DataStateType.training, length=0))
+        except DatasetScriptException:
+            global_leap_binder.setup_container = DatasetIntegrationSetup()
+            raise
         except TypeError as e:
             import traceback
             global_leap_binder.setup_container = DatasetIntegrationSetup()
@@ -89,6 +92,11 @@ class LeapLoader(LeapLoaderBase):
             sys.path.append(parent_path)
 
         file_path = Path(self.code_path, self.code_entry_name)
+        if not file_path.is_file():
+            raise DatasetScriptException(
+                f"Entry file not found: '{self.code_entry_name}' does not exist under '{self.code_path}'. "
+                f"Check the 'entryFile' setting in your leap.yaml."
+            )
         append_path_recursively(str(file_path))
 
         importlib.invalidate_caches()
@@ -103,6 +111,13 @@ class LeapLoader(LeapLoaderBase):
 
         spec.loader.exec_module(file)
 
+        if global_leap_binder.setup_container.preprocess is None:
+            raise DatasetScriptException(
+                f"No Tensorleap integration found in '{self.code_entry_name}': the script never called "
+                f"leap_binder.set_preprocess(...). Check the 'entryFile' setting in your leap.yaml — "
+                f"it may be pointing to the wrong file."
+            )
+
     @lru_cache()
     def metric_by_name(self) -> Dict[str, MetricHandlerData]:
         self.exec_script()

{code_loader-1.0.187 → code_loader-1.0.188.dev0}/pyproject.toml

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