workbench 0.8.185__tar.gz → 0.8.187__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: workbench
-Version: 0.8.185
+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.185 → workbench-0.8.187}/aws_setup/workbench_compute/workbench_compute/workbench_compute_stack.py

@@ -244,8 +244,6 @@ class WorkbenchComputeStack(Stack):
                         script_path = message['script_path']  # s3://bucket/path/to/script.py
                         size = message.get('size', 'small')
                         extra_env = message.get('environment', {})
-                        model_outputs = message.get('MODEL_OUTPUTS', [])
-                        endpoint_outputs = message.get('ENDPOINT_OUTPUTS', [])
 
                         script_name = Path(script_path).stem
                         job_name = f"workbench_{script_name}_{datetime.now():%Y%m%d_%H%M%S}"
@@ -260,14 +258,7 @@ class WorkbenchComputeStack(Stack):
                             *[{'name': k, 'value': v} for k, v in extra_env.items()]
                         ]
 
-                        # Add MODEL_OUTPUTS as comma-separated list if provided
-                        if model_outputs:
-                            env_vars.append({'name': 'MODEL_OUTPUTS', 'value': ','.join(model_outputs)})
-
-                        # Add ENDPOINT_OUTPUTS as comma-separated list if provided
-                        if endpoint_outputs:
-                            env_vars.append({'name': 'ENDPOINT_OUTPUTS', 'value': ','.join(endpoint_outputs)})
-
+                        # Submit Batch job to the queue
                         response = batch.submit_job(
                             jobName=job_name,
                             jobQueue=JOB_QUEUE,

{workbench-0.8.185 → 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.185 → 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/scripts/delete_data_capture.py

@@ -0,0 +1,72 @@
+#!/usr/bin/env python3
+import boto3
+import argparse
+from collections import defaultdict
+
+# Workbench Imports
+from workbench.utils.config_manager import ConfigManager
+
+
+def list_and_delete_data_capture(bucket, preview=True):
+    s3 = boto3.client("s3")
+    prefix = "endpoints/"
+
+    # Get all endpoint prefixes
+    paginator = s3.get_paginator("list_objects_v2")
+    endpoints = set()
+
+    for page in paginator.paginate(Bucket=bucket, Prefix=prefix, Delimiter="/"):
+        for common_prefix in page.get("CommonPrefixes", []):
+            endpoint_prefix = common_prefix["Prefix"]
+            if endpoint_prefix != prefix:
+                endpoints.add(endpoint_prefix)
+
+    # Count and optionally delete files per endpoint
+    counts = defaultdict(int)
+    to_delete = []
+
+    for endpoint_prefix in endpoints:
+        data_capture_prefix = f"{endpoint_prefix}data_capture/"
+
+        for page in paginator.paginate(Bucket=bucket, Prefix=data_capture_prefix):
+            for obj in page.get("Contents", []):
+                if obj["Key"].endswith(".jsonl"):
+                    endpoint_name = endpoint_prefix.split("/")[-2]
+                    counts[endpoint_name] += 1
+                    to_delete.append(obj["Key"])
+
+    # Display results
+    if counts:
+        print(f"\n{'Endpoint':<50} {'Files'}")
+        print("-" * 60)
+        for endpoint, count in sorted(counts.items()):
+            print(f"{endpoint:<50} {count:>6}")
+        print("-" * 60)
+        print(f"{'TOTAL':<50} {sum(counts.values()):>6}\n")
+    else:
+        print("No .jsonl files found in data_capture directories")
+        return
+
+    # Delete if not preview mode
+    if not preview:
+        print("Deleting files...")
+        # Delete in batches of 1000
+        for i in range(0, len(to_delete), 1000):
+            batch = to_delete[i : i + 1000]
+            s3.delete_objects(Bucket=bucket, Delete={"Objects": [{"Key": k} for k in batch]})
+        print(f"Deleted {len(to_delete)} files")
+    else:
+        print("PREVIEW MODE - no files deleted")
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description="Remove data capture files from SageMaker endpoints")
+    parser.add_argument("--delete", action="store_true", help="Actually delete files (default is preview)")
+    args = parser.parse_args()
+
+    # Get the default S3 bucket from Workbench config
+    config = ConfigManager()
+    bucket = config.get_config("WORKBENCH_BUCKET")
+    print(f"Using S3 bucket: {bucket}")
+
+    list_and_delete_data_capture(bucket, preview=not args.delete)

{workbench-0.8.185 → workbench-0.8.187}/src/workbench/core/artifacts/endpoint_core.py

@@ -993,9 +993,9 @@ class EndpointCore(Artifact):
         self.upsert_workbench_meta({"workbench_input": input})
 
     def delete(self):
-        """ "Delete an existing Endpoint: Underlying Models, Configuration, and Endpoint"""
+        """Delete an existing Endpoint: Underlying Models, Configuration, and Endpoint"""
         if not self.exists():
-            self.log.warning(f"Trying to delete an Model that doesn't exist: {self.name}")
+            self.log.warning(f"Trying to delete an Endpoint that doesn't exist: {self.name}")
 
         # Remove this endpoint from the list of registered endpoints
         self.log.info(f"Removing {self.name} from the list of registered endpoints...")

{workbench-0.8.185 → workbench-0.8.187}/src/workbench/core/artifacts/model_core.py

@@ -899,7 +899,7 @@ class ModelCore(Artifact):
     def delete(self):
         """Delete the Model Packages and the Model Group"""
         if not self.exists():
-            self.log.warning(f"Trying to delete an Model that doesn't exist: {self.name}")
+            self.log.warning(f"Trying to delete a Model that doesn't exist: {self.name}")
 
         # Call the Class Method to delete the Model Group
         ModelCore.managed_delete(model_group_name=self.name)

{workbench-0.8.185 → workbench-0.8.187}/src/workbench/model_scripts/uq_models/generated_model_script.py

@@ -19,7 +19,7 @@ from typing import List, Tuple
 # Template Placeholders
 TEMPLATE_PARAMS = {
     "target": "udm_asy_res_free_percent",
-    "features": ['naromatom', 'minabspartialcharge', 'bcut2d_mrhi', 'smr_vsa10', 'vsa_estate2', 'minpartialcharge', 'xpc_5d', 'sps', 'xc_3dv', 'smr_vsa7', 'bcut2d_logplow', 'mollogp', 'vsa_estate1', 'num_s_centers', 'vsa_estate4', 'peoe_vsa13', 'fr_nh2', 'bertzct', 'estate_vsa4', 'vsa_estate9', 'smr_vsa3', 'fr_nh1', 'molwt', 'estate_vsa5', 'slogp_vsa5', 'maxpartialcharge', 'estate_vsa1', 'fr_hoccn', 'xc_5d', 'nbase', 'chi1v', 'peoe_vsa10', 'tpsa', 'vsa_estate3', 'chi2v', 'estate_vsa8', 'numheteroatoms', 'estate_vsa2', 'peoe_vsa1', 'labuteasa', 'axp_4d', 'xch_7dv', 'chi0n', 'num_r_centers', 'vsa_estate8', 'minabsestateindex', 'bcut2d_chglo', 'bcut2d_mwhi', 'fr_nh0', 'chi4n', 'estate_vsa9', 'smr_vsa5', 'peoe_vsa2', 'peoe_vsa7', 'peoe_vsa9', 'kappa3', 'slogp_vsa3', 'fr_arn', 'estate_vsa3', 'avgipc', 'axp_5d', 'xpc_6d', 'c2sp2', 'peoe_vsa5', 'vsa_estate5', 'balabanj', 'maxabspartialcharge', 'fr_aniline', 'fr_piperdine', 'vsa_estate6', 'bcut2d_mwlow', 'numsaturatedheterocycles', 'vsa_estate10', 'smr_vsa1', 'estate_vsa6', 'smr_vsa6', 'fpdensitymorgan1', 'peoe_vsa3', 'peoe_vsa8', 'smr_vsa9', 'slogp_vsa2', 'nocount', 'fpdensitymorgan3', 'axp_6d', 'bcut2d_mrlow', 'bcut2d_logphi', 'axp_4dv', 'fpdensitymorgan2', 'mp', 'xp_5d', 'fr_nhpyrrole', 'mz', 'mv', 'vsa_estate7', 'axp_7dv', 'mi', 'c1sp2', 'xpc_6dv', 'slogp_vsa10', 'xp_7d', 'axp_3dv', 'peoe_vsa4', 'peoe_vsa6', 'axp_2dv', 'xch_5dv', 'qed', 'estate_vsa7', 'numaromaticrings', 'chi1n', 'axp_0d', 'axp_6dv', 'numrotatablebonds', 'hallkieralpha', 'c1sp3', 'xc_4dv', 'kappa2', 'bcut2d_chghi', 'xch_7d', 'axp_0dv', 'slogp_vsa7', 'axp_7d', 'minestateindex', 'axp_2d', 'axp_1d', 'chi0', 'fractioncsp3', 'slogp_vsa6', 'axp_1dv', 'chi2n', 'xp_6dv', 'maxestateindex', 'xpc_4d', 'numaliphaticheterocycles', 'chi1', 'phi', 'chi3n', 'xc_4d', 'xc_3d', 'peoe_vsa12', 'xp_6d', 'chi3v', 'axp_3d', 'axp_5dv', 'fr_benzene', 'slogp_vsa4', 'fr_pyridine', 'fr_aryl_methyl', 'xp_5dv', 'c3sp3', 'xp_7dv', 'slogp_vsa1', 'peoe_vsa11', 'mse', 'xc_5dv', 'xpc_5dv', 'xc_6dv', 'xp_0dv', 'xch_5d', 'c3sp2', 'numatomstereocenters', 'numhacceptors', 'fr_imidazole', 'numsaturatedrings', 'xpc_4dv', 'chi0v', 'numheterocycles', 'xch_6dv', 'estate_vsa10', 'chi4v', 'mare', 'numhdonors', 'xch_6d', 'xp_4d', 'fr_ar_n', 'numunspecifiedatomstereocenters', 'numspiroatoms', 'xch_4dv', 'fr_morpholine', 'fr_methoxy', 'mm', 'fr_piperzine'],
+    "features": ['naromatom', 'fr_nh2', 'mollogp', 'numheterocycles', 'bcut2d_mrhi', 'numaromaticrings', 'smr_vsa7', 'peoe_vsa4', 'slogp_vsa6', 'peoe_vsa8', 'vsa_estate3', 'maxabspartialcharge', 'fr_arn', 'bcut2d_logplow', 'chi1v', 'axp_6d', 'bcut2d_chglo', 'balabanj', 'slogp_vsa10', 'hallkieralpha', 'vsa_estate6', 'fpdensitymorgan1', 'sps', 'qed', 'peoe_vsa7', 'maxestateindex', 'estate_vsa8', 'vsa_estate9', 'fr_nhpyrrole', 'mz', 'mp', 'bcut2d_mwhi', 'peoe_vsa13', 'c2sp2', 'numrotatablebonds', 'kappa3', 'peoe_vsa1', 'slogp_vsa2', 'xc_5dv', 'bertzct', 'estate_vsa10', 'axp_0d', 'estate_vsa2', 'xc_4d', 'smr_vsa1', 'phi', 'estate_vsa3', 'vsa_estate2', 'mv', 'estate_vsa4', 'mm', 'fr_nh1', 'slogp_vsa7', 'chi4n', 'estate_vsa6', 'fpdensitymorgan2', 'molmr', 'mse', 'bcut2d_mwlow', 'bcut2d_mrlow', 'chi2v', 'minestateindex', 'xpc_4dv', 'fr_nh0', 'axp_2d', 'vsa_estate8', 'nhohcount', 'smr_vsa6', 'peoe_vsa9', 'smr_vsa5', 'num_r_centers', 'xpc_6dv', 'xc_3d', 'slogp_vsa5', 'axp_7dv', 'minabsestateindex', 'xc_5d', 'vsa_estate10', 'fr_hoccn', 'smr_vsa3', 'vsa_estate1', 'axp_5d', 'num_s_centers', 'axp_1d', 'estate_vsa1', 'fpdensitymorgan3', 'axp_5dv', 'chi3n', 'peoe_vsa6', 'labuteasa', 'chi2n', 'xc_6d', 'xp_7d', 'tpsa', 'xpc_4d', 'avgipc', 'xp_5d', 'vsa_estate5', 'xch_7d', 'xch_5d', 'axp_4dv', 'nbase', 'xc_3dv', 'kappa2', 'axp_3d', 'c1sp3', 'numhacceptors', 'bcut2d_logphi', 'smr_vsa10', 'fr_piperzine', 'peoe_vsa11', 'axp_6dv', 'peoe_vsa10', 'estate_vsa9', 'bcut2d_chghi', 'xp_6d', 'xch_6dv', 'chi0', 'vsa_estate7', 'mi', 'xpc_5d', 'fractioncsp3', 'xp_0dv', 'kappa1', 'minpartialcharge', 'xp_6dv', 'peoe_vsa2', 'chi3v', 'axp_0dv', 'mare', 'xch_5dv', 'vsa_estate4', 'xp_4dv', 'estate_vsa7', 'xp_3d', 'numaliphaticheterocycles', 'chi1', 'xp_3dv', 'fr_ether', 'xch_6d', 'peoe_vsa12', 'xch_7dv', 'axp_1dv', 'axp_7d', 'fr_ndealkylation2', 'smr_vsa9', 'axp_2dv', 'estate_vsa5', 'mpe', 'molwt', 'xch_4d', 'axp_3dv', 'xp_5dv', 'chi4v', 'heavyatommolwt', 'fr_al_oh', 'xpc_5dv', 'xpc_6d', 'maxpartialcharge', 'numatomstereocenters', 'peoe_vsa3', 'fr_aniline', 'minabspartialcharge', 'c3sp3', 'slogp_vsa1', 'exactmolwt', 'chi1n', 'xp_7dv', 'chi0n', 'xp_2d', 'xch_4dv', 'fr_bicyclic', 'xc_4dv', 'axp_4d', 'slogp_vsa4', 'fr_benzene', 'numaromaticheterocycles', 'fr_aryl_methyl', 'fr_pyridine', 'fr_imine', 'chi0v', 'slogp_vsa12'],
     "compressed_features": [],
     "train_all_data": True,
     "hyperparameters": {},
@@ -466,7 +466,7 @@ def predict_fn(df, models) -> pd.DataFrame:
     df["q_50"] = df["prediction"]
 
     # Calculate a pseudo-standard deviation from the 68% interval width
-    df["prediction_std"] = (df["q_84"] - df["q_16"]) / 2.0
+    df["prediction_std"] = (df["q_84"] - df["q_16"]).abs() / 2.0
 
     # Reorder the quantile columns for easier reading
     quantile_cols = ["q_025", "q_05", "q_10", "q_16", "q_25", "q_75", "q_84", "q_90", "q_95", "q_975"]

{workbench-0.8.185 → workbench-0.8.187}/src/workbench/model_scripts/uq_models/mapie.template

@@ -466,7 +466,7 @@ def predict_fn(df, models) -> pd.DataFrame:
     df["q_50"] = df["prediction"]
 
     # Calculate a pseudo-standard deviation from the 68% interval width
-    df["prediction_std"] = (df["q_84"] - df["q_16"]) / 2.0
+    df["prediction_std"] = (df["q_84"] - df["q_16"]).abs() / 2.0
 
     # Reorder the quantile columns for easier reading
     quantile_cols = ["q_025", "q_05", "q_10", "q_16", "q_25", "q_75", "q_84", "q_90", "q_95", "q_975"]

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.185 → workbench-0.8.187}/src/workbench/scripts/ml_pipeline_sqs.py

@@ -13,14 +13,21 @@ cm = ConfigManager()
 workbench_bucket = cm.get_config("WORKBENCH_BUCKET")
 
 
-def submit_to_sqs(script_path: str, size: str = "small", realtime: bool = False, recreate: bool = False) -> None:
+def submit_to_sqs(
+    script_path: str, size: str = "small", realtime: bool = False, dt: bool = False, promote: bool = False
+) -> None:
     """
     Upload script to S3 and submit message to SQS queue for processing.
+
     Args:
         script_path: Local path to the ML pipeline script
         size: Job size tier - "small" (default), "medium", or "large"
-        realtime: If True, sets serverless=False for real-time processing (default: False, meaning serverless=True)
-        recreate: If True, sets RECREATE=True in environment (default: False)
+        realtime: If True, sets serverless=False for real-time processing (default: False)
+        dt: If True, sets DT=True in environment (default: False)
+        promote: If True, sets PROMOTE=True in environment (default: False)
+
+    Raises:
+        ValueError: If size is invalid or script file not found
     """
     print(f"\n{'=' * 60}")
     print("🚀  SUBMITTING ML PIPELINE JOB")
@@ -36,7 +43,8 @@ def submit_to_sqs(script_path: str, size: str = "small", realtime: bool = False,
     print(f"📄  Script: {script_file.name}")
     print(f"📏  Size tier: {size}")
     print(f"⚡  Mode: {'Real-time' if realtime else 'Serverless'} (serverless={'False' if realtime else 'True'})")
-    print(f"🔄  Recreate: {recreate}")
+    print(f"🔄  DynamicTraining: {dt}")
+    print(f"🆕  Promote: {promote}")
     print(f"🪣  Bucket: {workbench_bucket}")
     sqs = AWSAccountClamp().boto3_session.client("sqs")
     script_name = script_file.name
@@ -94,14 +102,15 @@ def submit_to_sqs(script_path: str, size: str = "small", realtime: bool = False,
     message = {"script_path": s3_path, "size": size}
 
     # Set environment variables
-    message["environment"] = {"SERVERLESS": "False" if realtime else "True"}
-    if recreate:
-        message["environment"]["RECREATE"] = "True"
-
-    print("\n📨  Sending message to SQS...")
+    message["environment"] = {
+        "SERVERLESS": "False" if realtime else "True",
+        "DT": str(dt),
+        "PROMOTE": str(promote),
+    }
 
     # Send the message to SQS
     try:
+        print("\n📨  Sending message to SQS...")
         response = sqs.send_message(
             QueueUrl=queue_url,
             MessageBody=json.dumps(message, indent=2),
@@ -121,7 +130,8 @@ def submit_to_sqs(script_path: str, size: str = "small", realtime: bool = False,
     print(f"📄  Script: {script_name}")
     print(f"📏  Size: {size}")
     print(f"⚡  Mode: {'Real-time' if realtime else 'Serverless'} (SERVERLESS={'False' if realtime else 'True'})")
-    print(f"🔄  Recreate: {recreate}")
+    print(f"🔄  DynamicTraining: {dt}")
+    print(f"🆕  Promote: {promote}")
     print(f"🆔  Message ID: {message_id}")
     print("\n🔍  MONITORING LOCATIONS:")
     print(f"   • SQS Queue: AWS Console → SQS → {queue_name}")
@@ -144,13 +154,18 @@ def main():
         help="Create realtime endpoints (default is serverless)",
     )
     parser.add_argument(
-        "--recreate",
+        "--dt",
+        action="store_true",
+        help="Set DT=True (models and endpoints will have '-dt' suffix)",
+    )
+    parser.add_argument(
+        "--promote",
         action="store_true",
-        help="Set RECREATE=True (will force recreation of resources)",
+        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, recreate=args.recreate)
+        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.185 → 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.185 → 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.185 → workbench-0.8.187}/src/workbench.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: workbench
-Version: 0.8.185
+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.185 → workbench-0.8.187}/src/workbench.egg-info/SOURCES.txt

@@ -377,6 +377,7 @@ scripts/active_logs.sh
 scripts/ag_table_row_selection.py
 scripts/compare_dataframes.py
 scripts/compare_model_packages.py
+scripts/delete_data_capture.py
 scripts/endpoint_timing.py
 scripts/find_smiles_issues.py
 scripts/get_tags.py
@@ -594,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
@@ -655,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.185 → 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.185 → 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.185/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())