code-loader 0.2.61__tar.gz → 0.2.63__tar.gz

{code_loader-0.2.61 → code_loader-0.2.63}/PKG-INFO

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "code-loader"
-version = "0.2.61"
+version = "0.2.63"
 description = ""
 authors = ["dorhar <doron.harnoy@tensorleap.ai>"]
 license = "MIT"
@@ -17,7 +17,6 @@ numpy = "^1.22.3"
 tensorflow = {version = "^2.11.0", markers = "platform_machine  == 'x86_64'"}
 tensorflow-macos = {version = "^2.11.0", markers = "platform_machine  == 'arm64'"}
 typeguard = "^2.13.3"
-leap-model-parser = "0.1.86"
 
 [tool.poetry.dev-dependencies]
 pytest = "^7.1.1"

code_loader-0.2.61/setup.py

@@ -1,44 +0,0 @@
-# -*- coding: utf-8 -*-
-from setuptools import setup
-
-packages = \
-['code_loader',
- 'code_loader.contract',
- 'code_loader.helpers',
- 'code_loader.helpers.detection',
- 'code_loader.helpers.detection.yolo',
- 'code_loader.leap_binder',
- 'code_loader.metrics',
- 'code_loader.visualizers']
-
-package_data = \
-{'': ['*']}
-
-install_requires = \
-['leap-model-parser==0.1.86',
- 'numpy>=1.22.3,<2.0.0',
- 'typeguard>=2.13.3,<3.0.0']
-
-extras_require = \
-{':platform_machine == "arm64"': ['tensorflow-macos>=2.11.0,<3.0.0'],
- ':platform_machine == "x86_64"': ['tensorflow>=2.11.0,<3.0.0']}
-
-setup_kwargs = {
-    'name': 'code-loader',
-    'version': '0.2.61',
-    'description': '',
-    'long_description': '# tensorleap code loader\nUsed to load user code to tensorleap \n',
-    'author': 'dorhar',
-    'author_email': 'doron.harnoy@tensorleap.ai',
-    'maintainer': 'None',
-    'maintainer_email': 'None',
-    'url': 'https://github.com/tensorleap/code-loader',
-    'packages': packages,
-    'package_data': package_data,
-    'install_requires': install_requires,
-    'extras_require': extras_require,
-    'python_requires': '>=3.8,<3.11',
-}
-
-
-setup(**setup_kwargs)