workbench 0.8.186__tar.gz → 0.8.187__tar.gz

{workbench-0.8.186 → workbench-0.8.187}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: workbench
-Version: 0.8.186
+Version: 0.8.187
 Summary: Workbench: A Dashboard and Python API for creating and deploying AWS SageMaker Model Pipelines
 Author-email: SuperCowPowers LLC <support@supercowpowers.com>
 License-Expression: MIT
@@ -10,6 +10,7 @@ Classifier: Development Status :: 4 - Beta
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Scientific/Engineering
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

{workbench-0.8.186 → workbench-0.8.187}/model_docker_images/ml_pipelines/Dockerfile

@@ -29,7 +29,7 @@ COPY --from=builder /usr/local/lib/python3.12/site-packages /usr/local/lib/pytho
 COPY --from=builder /usr/local/bin /usr/local/bin
 
 # Install workbench only
-RUN pip install --no-cache-dir workbench==0.8.182
+RUN pip install --no-cache-dir workbench==0.8.186
 
 # Add the script runner
 COPY ml_pipeline_runner.py /app/ml_pipeline_runner.py

{workbench-0.8.186 → workbench-0.8.187}/pyproject.toml

@@ -19,6 +19,7 @@ classifiers = [
     "Programming Language :: Python :: 3.10",
     "Programming Language :: Python :: 3.11",
     "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
     "Topic :: Scientific/Engineering",
 ]
 authors = [{ name = "SuperCowPowers LLC", email = "support@supercowpowers.com" }]
@@ -92,6 +93,7 @@ workbench_config = "workbench.scripts.show_config:main"
 glue_launcher = "workbench.scripts.glue_launcher:main"
 ml_pipeline_batch = "workbench.scripts.ml_pipeline_batch:main"
 ml_pipeline_sqs = "workbench.scripts.ml_pipeline_sqs:main"
+lambda_launcher = "workbench.scripts.lambda_launcher:main"
 
 [tool.pytest.ini_options]
 addopts = "-v --cov-report term-missing"

workbench-0.8.187/src/workbench/scripts/lambda_launcher.py

@@ -0,0 +1,39 @@
+import sys
+import os
+import importlib.util
+
+
+def main():
+    if len(sys.argv) != 2:
+        print("Usage: lambda_launcher <handler_module_name>")
+        sys.exit(1)
+
+    handler_file = sys.argv[1]
+
+    # Add .py if not present
+    if not handler_file.endswith(".py"):
+        handler_file += ".py"
+
+    # Check if file exists
+    if not os.path.exists(handler_file):
+        print(f"Error: File '{handler_file}' not found")
+        sys.exit(1)
+
+    # Load the module dynamically
+    spec = importlib.util.spec_from_file_location("lambda_module", handler_file)
+    lambda_module = importlib.util.module_from_spec(spec)
+    spec.loader.exec_module(lambda_module)
+
+    # Call the lambda_handler
+    print(f"Invoking lambda_handler from {handler_file}...")
+    print("-" * 50)
+
+    result = lambda_module.lambda_handler({}, {})
+
+    print("-" * 50)
+    print("Result:")
+    print(result)
+
+
+if __name__ == "__main__":
+    main()

{workbench-0.8.186 → workbench-0.8.187}/src/workbench/scripts/ml_pipeline_sqs.py

@@ -158,9 +158,14 @@ def main():
         action="store_true",
         help="Set DT=True (models and endpoints will have '-dt' suffix)",
     )
+    parser.add_argument(
+        "--promote",
+        action="store_true",
+        help="Set Promote=True (models and endpoints will use promoted naming",
+    )
     args = parser.parse_args()
     try:
-        submit_to_sqs(args.script_file, args.size, realtime=args.realtime, dt=args.dt)
+        submit_to_sqs(args.script_file, args.size, realtime=args.realtime, dt=args.dt, promote=args.promote)
     except Exception as e:
         print(f"\n❌  ERROR: {e}")
         log.error(f"Error: {e}")

{workbench-0.8.186 → workbench-0.8.187}/src/workbench/utils/config_manager.py

@@ -4,16 +4,13 @@ import os
 import sys
 import platform
 import logging
-import importlib.resources as resources  # noqa: F401 Python 3.9 compatibility
 from typing import Any, Dict
+from importlib.resources import files, as_file
 
 # Workbench imports
 from workbench.utils.license_manager import LicenseManager
 from workbench_bridges.utils.execution_environment import running_as_service
 
-# Python 3.9 compatibility
-from workbench.utils.resource_utils import get_resource_path
-
 
 class FatalConfigError(Exception):
     """Exception raised for errors in the configuration."""
@@ -172,8 +169,7 @@ class ConfigManager:
         Returns:
             str: The open source API key.
         """
-        # Python 3.9 compatibility
-        with get_resource_path("workbench.resources", "open_source_api.key") as open_source_key_path:
+        with as_file(files("workbench.resources").joinpath("open_source_api.key")) as open_source_key_path:
             with open(open_source_key_path, "r") as key_file:
                 return key_file.read().strip()
 

{workbench-0.8.186 → workbench-0.8.187}/src/workbench/utils/license_manager.py

@@ -6,15 +6,12 @@ import json
 import logging
 import requests
 from typing import Union
-import importlib.resources as resources  # noqa: F401 Python 3.9 compatibility
 from datetime import datetime
 from cryptography.hazmat.primitives import hashes
 from cryptography.hazmat.primitives.asymmetric import padding
 from cryptography.hazmat.primitives import serialization
 from cryptography.hazmat.backends import default_backend
-
-# Python 3.9 compatibility
-from workbench.utils.resource_utils import get_resource_path
+from importlib.resources import files, as_file
 
 
 class FatalLicenseError(Exception):
@@ -140,8 +137,7 @@ class LicenseManager:
         Returns:
             The public key as an object.
         """
-        # Python 3.9 compatibility
-        with get_resource_path("workbench.resources", "signature_verify_pub.pem") as public_key_path:
+        with as_file(files("workbench.resources").joinpath("signature_verify_pub.pem")) as public_key_path:
             with open(public_key_path, "rb") as key_file:
                 public_key_data = key_file.read()
 

{workbench-0.8.186 → workbench-0.8.187}/src/workbench.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: workbench
-Version: 0.8.186
+Version: 0.8.187
 Summary: Workbench: A Dashboard and Python API for creating and deploying AWS SageMaker Model Pipelines
 Author-email: SuperCowPowers LLC <support@supercowpowers.com>
 License-Expression: MIT
@@ -10,6 +10,7 @@ Classifier: Development Status :: 4 - Beta
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Scientific/Engineering
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

{workbench-0.8.186 → workbench-0.8.187}/src/workbench.egg-info/SOURCES.txt

@@ -595,6 +595,7 @@ src/workbench/resources/open_source_api.key
 src/workbench/resources/signature_verify_pub.pem
 src/workbench/scripts/check_double_bond_stereo.py
 src/workbench/scripts/glue_launcher.py
+src/workbench/scripts/lambda_launcher.py
 src/workbench/scripts/ml_pipeline_batch.py
 src/workbench/scripts/ml_pipeline_sqs.py
 src/workbench/scripts/monitor_cloud_watch.py
@@ -656,7 +657,6 @@ src/workbench/utils/plugin_manager.py
 src/workbench/utils/prox_utils.py
 src/workbench/utils/redis_cache.py
 src/workbench/utils/repl_utils.py
-src/workbench/utils/resource_utils.py
 src/workbench/utils/s3_utils.py
 src/workbench/utils/shap_utils.py
 src/workbench/utils/shapley_values.py

{workbench-0.8.186 → workbench-0.8.187}/src/workbench.egg-info/entry_points.txt

@@ -1,6 +1,7 @@
 [console_scripts]
 cloud_watch = workbench.scripts.monitor_cloud_watch:main
 glue_launcher = workbench.scripts.glue_launcher:main
+lambda_launcher = workbench.scripts.lambda_launcher:main
 ml_pipeline_batch = workbench.scripts.ml_pipeline_batch:main
 ml_pipeline_sqs = workbench.scripts.ml_pipeline_sqs:main
 workbench = workbench.repl.workbench_shell:launch_shell

{workbench-0.8.186 → workbench-0.8.187}/tox.ini

@@ -1,6 +1,6 @@
 [tox]
 requires = tox>=4.24.0
-env_list = lint, py312
+env_list = lint, py313
 
 [testenv]
 description = Run unit tests
@@ -21,11 +21,6 @@ description = Run tests with verbose logging
 deps = {[testenv]deps}
 commands = pytest --assert=plain -x -m "not long" -s {posargs:tests}
 
-[testenv:py39]
-description = Run tests with Python 3.9
-basepython = python3.9
-deps = {[testenv]deps}
-
 [testenv:py310]
 description = Run tests with Python 3.10
 basepython = python3.10
@@ -41,6 +36,16 @@ description = Run tests with Python 3.12
 basepython = python3.12
 deps = {[testenv]deps}
 
+[testenv:py313]
+description = Run tests with Python 3.13
+basepython = python3.13
+deps = {[testenv]deps}
+
+[testenv:py314]
+description = Run tests with Python 3.14
+basepython = python3.14
+deps = {[testenv]deps}
+
 [testenv:long]
 description = Run long-duration tests
 setenv = WORKBENCH_DEBUG = True
@@ -65,4 +70,4 @@ commands = black --line-length=120 {posargs:.}
 [testenv:all]
 description = Run all environments
 deps = tox>=4
-commands = tox -e lint -e py310 -e py311 -e py312 -e long
+commands = tox -e lint -e py310 -e py311 -e py312 -e py313 -e long

workbench-0.8.186/src/workbench/utils/resource_utils.py

@@ -1,39 +0,0 @@
-"""Resource utilities for Workbench"""
-
-import sys
-import importlib.resources as resources
-import pathlib
-import pkg_resources
-
-
-def get_resource_path(package: str, resource: str) -> pathlib.Path:
-    """Get the path to a resource file, compatible with Python 3.9 and higher.
-
-    Args:
-        package (str): The package where the resource is located.
-        resource (str): The name of the resource file.
-
-    Returns:
-        pathlib.Path: The path to the resource file.
-    """
-    if sys.version_info >= (3, 10):
-        # Python 3.10 and higher: use importlib.resources.path
-        with resources.path(package, resource) as path:
-            return path
-    else:
-        # Python 3.9 and lower: manually construct the path based on package location
-        # Get the location of the installed package
-        package_location = pathlib.Path(pkg_resources.get_distribution(package.split(".")[0]).location)
-        resource_path = package_location / package.replace(".", "/") / resource
-
-        if resource_path.exists():
-            return resource_path
-        else:
-            raise FileNotFoundError(f"Resource '{resource}' not found in package '{package}'.")
-
-
-if __name__ == "__main__":
-    # Test the resource utilities
-    with get_resource_path("workbench.resources", "open_source_api.key") as open_source_key_path:
-        with open(open_source_key_path, "r") as key_file:
-            print(key_file.read().strip())