code-loader 0.2.61__py3-none-any.whl → 0.2.63__py3-none-any.whl

code_loader/leaploader.py

@@ -1,8 +1,10 @@
 import importlib.util
+import io
+import sys
+from contextlib import redirect_stdout
 from functools import lru_cache
 from pathlib import Path
-from typing import Dict, List, Iterable, Any, Union
-import sys
+from typing import Dict, List, Iterable, Any, Union, Optional
 
 import numpy as np
 import numpy.typing as npt
@@ -118,22 +120,33 @@ class LeapLoader:
         test_payloads: List[DatasetTestResultPayload] = []
         setup_response = None
         general_error = None
-        try:
-            self.exec_script()
-            preprocess_test_payload = self._check_preprocess()
-            test_payloads.append(preprocess_test_payload)
-            handlers_test_payloads = self._check_handlers()
-            test_payloads.extend(handlers_test_payloads)
-            is_valid = all([payload.is_passed for payload in test_payloads])
-            setup_response = self.get_dataset_setup_response()
-        except DatasetScriptException as e:
-            line_number = get_root_exception_line_number()
-            general_error = f"Something went wrong, {repr(e.__cause__)} line number: {line_number}"
-            is_valid = False
-        except Exception as e:
-            line_number = get_root_exception_line_number()
-            general_error = f"Something went wrong, {repr(e)} line number: {line_number}"
-            is_valid = False
+        stdout_steam = io.StringIO()
+        with redirect_stdout(stdout_steam):
+            try:
+                self.exec_script()
+                preprocess_test_payload = self._check_preprocess()
+                test_payloads.append(preprocess_test_payload)
+                handlers_test_payloads = self._check_handlers()
+                test_payloads.extend(handlers_test_payloads)
+                is_valid = all([payload.is_passed for payload in test_payloads])
+                setup_response = self.get_dataset_setup_response()
+            except DatasetScriptException as e:
+                line_number = get_root_exception_line_number()
+                general_error = f"Something went wrong, {repr(e.__cause__)} line number: {line_number}"
+                is_valid = False
+            except Exception as e:
+                line_number = get_root_exception_line_number()
+                general_error = f"Something went wrong, {repr(e)} line number: {line_number}"
+                is_valid = False
+
+        print_log = stdout_steam.getvalue()
+        if print_log is not None and print_log != "":
+            if general_error is None:
+                general_error = ""
+            else:
+                general_error += "\n\n"
+            general_error += "Print Log:\n"
+            general_error += print_log
 
         is_valid_for_model = bool(global_leap_binder.setup_container.custom_layers)
         model_setup = self.get_model_setup_response()

{code_loader-0.2.61.dist-info → code_loader-0.2.63.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: code-loader
-Version: 0.2.61
+Version: 0.2.63
 Summary: 
 Home-page: https://github.com/tensorleap/code-loader
 License: MIT
@@ -12,10 +12,9 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
-Requires-Dist: leap-model-parser (==0.1.86)
 Requires-Dist: numpy (>=1.22.3,<2.0.0)
-Requires-Dist: tensorflow (>=2.11.0,<3.0.0); platform_machine == "x86_64"
-Requires-Dist: tensorflow-macos (>=2.11.0,<3.0.0); platform_machine == "arm64"
+Requires-Dist: tensorflow (>=2.11.0,<3.0.0) ; platform_machine == "x86_64"
+Requires-Dist: tensorflow-macos (>=2.11.0,<3.0.0) ; platform_machine == "arm64"
 Requires-Dist: typeguard (>=2.13.3,<3.0.0)
 Project-URL: Repository, https://github.com/tensorleap/code-loader
 Description-Content-Type: text/markdown

{code_loader-0.2.61.dist-info → code_loader-0.2.63.dist-info}/RECORD

@@ -17,13 +17,13 @@ code_loader/helpers/detection/yolo/pytorch_utils.py,sha256=D4Rreaehz3R8YMvLRbPIh
 code_loader/helpers/detection/yolo/utils.py,sha256=eV9-IZhglRNGud1UmSckIuA49WRKJitDQhPKK7iIaBc,6563
 code_loader/leap_binder/__init__.py,sha256=Oe7DNj6RmgX-tMruOrI7GYvf3SyAnzOxugF2e117_Z8,93
 code_loader/leap_binder/leapbinder.py,sha256=WjC-UA5xeY9UZtMqJQOxjidlFcWURC9mmy7OrQy1IY0,6692
-code_loader/leaploader.py,sha256=3Ux6NHd1TUYUf871F55E7sFL3nv4bUOG16c610--7bY,15207
+code_loader/leaploader.py,sha256=2p7db8DPeMUdnf3ptp5WlYYQRE6m4CMF9uLJ9xQS86w,15724
 code_loader/metrics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/metrics/default_metrics.py,sha256=xtvTskR7LC5pyKSlFcvlyBpFY9TSXNGVAi5taEPSI60,6727
 code_loader/utils.py,sha256=WUcM97OuxrhfLCRPoH9EbXrxajNpYgX1CTMc3_PXtYU,1736
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=HqWx2qfTrroGl2n8Fpmr_4X-rk7tE2oGapjO3gzz4WY,2226
-code_loader-0.2.61.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-0.2.61.dist-info/WHEEL,sha256=vxFmldFsRN_Hx10GDvsdv1wroKq8r5Lzvjp6GZ4OO8c,88
-code_loader-0.2.61.dist-info/METADATA,sha256=_ObkCGWAmO5QGYOBYUYDOhE5MCghQJs4wrMUyLAU0Zg,918
-code_loader-0.2.61.dist-info/RECORD,,
+code_loader-0.2.63.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-0.2.63.dist-info/METADATA,sha256=fXPeCm9x-eYgu_4TVN3a_H64OZa4zdG9oouBJI7TQ_k,876
+code_loader-0.2.63.dist-info/WHEEL,sha256=kLuE8m1WYU0Ig0_YEGrXyTtiJvKPpLpDEiChiNyei5Y,88
+code_loader-0.2.63.dist-info/RECORD,,

{code_loader-0.2.61.dist-info → code_loader-0.2.63.dist-info}/WHEEL

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