code-loader 1.0.187__py3-none-any.whl → 1.0.188.dev0__py3-none-any.whl

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

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

@@ -22,8 +22,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=XLYYcV50qjMvoC1S6WW0tLBch_0g5gl1UyHiVSWYbvg,40491
-code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=8ob0KdvpeeNEzA3OZ4UL3dU3gigjB959zNfSvB8KHzw,118570
-code_loader/leaploader.py,sha256=K9Q5kiCZ_A2GkS5qOEantAMvWGaz3bmO1X8buXDSpgg,44682
+code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=qfKa--0kG_K_Eb8wRd5tIXEJ6rbKva3360gU8EU9eKo,118375
+code_loader/leaploader.py,sha256=qtWn_6EsKuerCcURitCsFzEk2MSoRDUlHHBM0ohI9Io,45472
 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 +32,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.187.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.187.dist-info/METADATA,sha256=uVGw15k07GJwz-s1QIHhL6IUkfbfDv49kUaD6YlceQI,1090
-code_loader-1.0.187.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-code_loader-1.0.187.dist-info/RECORD,,
+code_loader-1.0.188.dev0.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.188.dev0.dist-info/METADATA,sha256=mtdX0lb1AbMGSDjxqIesGwlp3GNQoYFBcSBhIdaB8Lk,1095
+code_loader-1.0.188.dev0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+code_loader-1.0.188.dev0.dist-info/RECORD,,