workbench 0.8.165__py3-none-any.whl → 0.8.166__py3-none-any.whl

workbench/api/meta.py

@@ -113,13 +113,16 @@ class Meta(CloudMeta):
         """
         return super().models(details=details)
 
-    def endpoints(self) -> pd.DataFrame:
+    def endpoints(self, details: bool = False) -> pd.DataFrame:
         """Get a summary of the Endpoints deployed in the Cloud Platform
 
+        Args:
+            details (bool, optional): Include detailed information. Defaults to False.
+
         Returns:
             pd.DataFrame: A summary of the Endpoints in the Cloud Platform
         """
-        return super().endpoints()
+        return super().endpoints(details=details)
 
     def pipelines(self) -> pd.DataFrame:
         """Get a summary of the ML Pipelines deployed in the Cloud Platform

workbench/core/artifacts/monitor_core.py

@@ -2,6 +2,8 @@
 
 import logging
 import json
+import re
+from datetime import datetime
 from typing import Union, Tuple
 import pandas as pd
 from sagemaker import Predictor
@@ -283,67 +285,66 @@ class MonitorCore:
             self.log.error(f"Error checking data capture percentage: {e}")
             return None
 
-    def get_captured_data(self, max_files=None, add_timestamp=True) -> Tuple[pd.DataFrame, pd.DataFrame]:
+    def get_captured_data(self, from_date=None, add_timestamp=True) -> Tuple[pd.DataFrame, pd.DataFrame]:
         """
         Read and process captured data from S3.
 
         Args:
-            max_files (int, optional): Maximum number of files to process.
+            from_date (str, optional): Only process files from this date onwards (YYYY-MM-DD format).
                                        Defaults to None to process all files.
             add_timestamp (bool, optional): Whether to add a timestamp column to the DataFrame.
 
         Returns:
             Tuple[pd.DataFrame, pd.DataFrame]: Processed input and output DataFrames.
         """
-        # List files in the specified S3 path
         files = wr.s3.list_objects(self.data_capture_path)
         if not files:
             self.log.warning(f"No data capture files found in {self.data_capture_path}.")
             return pd.DataFrame(), pd.DataFrame()
 
-        self.log.info(f"Found {len(files)} files in {self.data_capture_path}.")
-
-        # Sort files by timestamp (assuming the naming convention includes timestamp)
+        # Filter by date if specified
+        if from_date:
+            from_date_obj = datetime.strptime(from_date, "%Y-%m-%d").date()
+            files = [f for f in files if self._file_date_filter(f, from_date_obj)]
+            self.log.info(f"Processing {len(files)} files from {from_date} onwards.")
+        else:
+            self.log.info(f"Processing all {len(files)} files.")
         files.sort()
 
-        # Select files to process
-        if max_files is None:
-            files_to_process = files
-            self.log.info(f"Processing all {len(files)} files.")
-        else:
-            files_to_process = files[-max_files:] if files else []
-            self.log.info(f"Processing the {len(files_to_process)} most recent file(s).")
-
-        # Process each file
-        all_input_dfs = []
-        all_output_dfs = []
-        for file_path in files_to_process:
-            self.log.info(f"Processing {file_path}...")
+        # Process files
+        all_input_dfs, all_output_dfs = [], []
+        for file_path in files:
             try:
-                # Read the JSON lines file
                 df = wr.s3.read_json(path=file_path, lines=True)
                 if not df.empty:
                     input_df, output_df = process_data_capture(df)
-                    # Generate a timestamp column if requested
                     if add_timestamp:
-                        # Get file metadata to extract last modified time
-                        file_metadata = wr.s3.describe_objects(path=file_path)
-                        timestamp = file_metadata[file_path]["LastModified"]
+                        timestamp = wr.s3.describe_objects(path=file_path)[file_path]["LastModified"]
                         output_df["timestamp"] = timestamp
-
-                    # Append the processed DataFrames to the lists
                     all_input_dfs.append(input_df)
                     all_output_dfs.append(output_df)
             except Exception as e:
-                self.log.warning(f"Error processing file {file_path}: {e}")
+                self.log.warning(f"Error processing {file_path}: {e}")
 
-        # Combine all DataFrames
-        if not all_input_dfs or not all_output_dfs:
-            self.log.warning("No valid data was processed from the captured files.")
+        if not all_input_dfs:
+            self.log.warning("No valid data was processed.")
             return pd.DataFrame(), pd.DataFrame()
 
         return pd.concat(all_input_dfs, ignore_index=True), pd.concat(all_output_dfs, ignore_index=True)
 
+    def _file_date_filter(self, file_path, from_date_obj):
+        """Extract date from S3 path and compare with from_date."""
+        try:
+            # Match YYYY/MM/DD pattern in the path
+            date_match = re.search(r"/(\d{4})/(\d{2})/(\d{2})/", file_path)
+            if date_match:
+                year, month, day = date_match.groups()
+                file_date = datetime(int(year), int(month), int(day)).date()
+                return file_date >= from_date_obj
+            return False  # No date pattern found
+        except ValueError:
+            return False
+
     def baseline_exists(self) -> bool:
         """
         Check if baseline files exist in S3.

workbench/core/cloud_platform/aws/aws_meta.py

@@ -308,7 +308,7 @@ class AWSMeta:
                     "Status": endpoint_details.get("EndpointStatus", "-"),
                     "Config": endpoint_details.get("EndpointConfigName", "-"),
                     "Variant": endpoint_details["config"]["variant"],
-                    "Capture": str(endpoint_details.get("DataCaptureConfig", {}).get("EnableCapture", "False")),
+                    "Capture": str(endpoint_details.get("DataCaptureConfig", {}).get("EnableCapture", "-")),
                     "Samp(%)": str(endpoint_details.get("DataCaptureConfig", {}).get("CurrentSamplingPercentage", "-")),
                     "Tags": aws_tags.get("workbench_tags", "-"),
                     "Monitored": endpoint_details["monitored"],

workbench/core/cloud_platform/aws/aws_session.py

@@ -10,7 +10,7 @@ import logging
 
 # Workbench Imports
 from workbench.utils.config_manager import ConfigManager
-from workbench.utils.execution_environment import running_on_lambda, running_on_glue
+from workbench_bridges.utils.execution_environment import running_on_lambda, running_on_glue
 
 # Attempt to import IPython-related utilities
 try:

workbench/repl/workbench_shell.py

@@ -72,7 +72,7 @@ if not ConfigManager().config_okay():
 
 # Set the log level to important
 log = logging.getLogger("workbench")
-log.setLevel(IMPORTANT_LEVEL_NUM)
+log.setLevel(logging.INFO)
 log.addFilter(
     lambda record: not (
         record.getMessage().startswith("Async: Metadata") or record.getMessage().startswith("Updated Metadata")

workbench/utils/aws_utils.py

@@ -55,7 +55,8 @@ def aws_throttle(func=None, retry_intervals=None):
     if func is None:
         return lambda f: aws_throttle(f, retry_intervals=retry_intervals)
 
-    service_hold_time = 2  # Seconds to wait before calling AWS function
+    # This is currently commented out (we might want to use it later)
+    # service_hold_time = 2  # Seconds to wait before calling AWS function
     default_intervals = [2**i for i in range(1, 9)]  # Default exponential backoff: 2, 4, 8... 256 seconds
     intervals = retry_intervals or default_intervals
 
@@ -64,8 +65,8 @@ def aws_throttle(func=None, retry_intervals=None):
         for attempt, delay in enumerate(intervals, start=1):
             try:
                 # Add sleep before calling AWS func if running as a service
-                if cm.running_as_service:
-                    time.sleep(service_hold_time)
+                # if cm.running_as_service:
+                #    time.sleep(service_hold_time)
                 return func(*args, **kwargs)
             except ClientError as e:
                 if e.response["Error"]["Code"] == "ThrottlingException":

workbench/utils/cloudwatch_handler.py

@@ -5,7 +5,7 @@ import getpass
 import time  # For managing send intervals
 
 # Workbench imports
-from workbench.utils.execution_environment import (
+from workbench_bridges.utils.execution_environment import (
     running_on_lambda,
     running_on_glue,
     running_on_ecs,

workbench/utils/config_manager.py

@@ -9,7 +9,7 @@ from typing import Any, Dict
 
 # Workbench imports
 from workbench.utils.license_manager import LicenseManager
-from workbench.utils.execution_environment import running_as_service
+from workbench_bridges.utils.execution_environment import running_as_service
 
 # Python 3.9 compatibility
 from workbench.utils.resource_utils import get_resource_path

{workbench-0.8.165.dist-info → workbench-0.8.166.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: workbench
-Version: 0.8.165
+Version: 0.8.166
 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
@@ -30,7 +30,7 @@ Requires-Dist: joblib>=1.3.2
 Requires-Dist: requests>=2.26.0
 Requires-Dist: rdkit>=2024.9.5
 Requires-Dist: mordredcommunity>=2.0.6
-Requires-Dist: workbench-bridges>=0.1.8
+Requires-Dist: workbench-bridges>=0.1.10
 Provides-Extra: ui
 Requires-Dist: plotly>=6.0.0; extra == "ui"
 Requires-Dist: dash>3.0.0; extra == "ui"

{workbench-0.8.165.dist-info → workbench-0.8.166.dist-info}/RECORD

@@ -34,7 +34,7 @@ workbench/api/df_store.py,sha256=Wybb3zO-jPpAi2Ns8Ks1-lagvXAaBlRpBZHhnnl3Lms,613
 workbench/api/endpoint.py,sha256=RWGqxsCW_pMiENMb_XZlm2ZCldMS4suEBM3F5gT3hYI,3814
 workbench/api/feature_set.py,sha256=wzNxNjN0K2FaIC7QUIogMnoHqw2vo0iAHYlGk6fWLCw,6649
 workbench/api/graph_store.py,sha256=LremJyPrQFgsHb7hxsctuCsoxx3p7TKtaY5qALHe6pc,4372
-workbench/api/meta.py,sha256=fCOtZMfAHWaerzcsTeFnimXfgV8STe9JDiB7QBogktc,8456
+workbench/api/meta.py,sha256=1_9989cPvf3hd3tA-83hLijOGNnhwXAF8aZF45adeDQ,8596
 workbench/api/model.py,sha256=2hPN8UK4whZ0kDgPtbR7lEknw7XhH5hGYaHA55jmZWQ,4529
 workbench/api/monitor.py,sha256=kQHSFiVLRWnHekSdatMKR3QbRj1BBNrVXpZgvV83LPM,5027
 workbench/api/parameter_store.py,sha256=7BObkuATuP6C5AG_46kCWsmuCwuh1vgMJDBSN0gTkwM,4294
@@ -56,16 +56,16 @@ workbench/core/artifacts/data_source_factory.py,sha256=YL_tA5fsgubbB3dPF6T4tO0rG
 workbench/core/artifacts/endpoint_core.py,sha256=6uDOl-VKrTbLMlHZEYFY80XwrCP5H0W36JoHySjhl7M,48163
 workbench/core/artifacts/feature_set_core.py,sha256=055VdSYR09HP4ygAuYvIYtHQ7Ec4XxsZygpgEl5H5jQ,29136
 workbench/core/artifacts/model_core.py,sha256=U0dSkpZMrsIgbUglVkPwAgN0gji7Oa7glOjqMQJDAzE,50927
-workbench/core/artifacts/monitor_core.py,sha256=BP6UuCyBI4zB2wwcIXvUw6RC0EktTcQd5Rv0x73qzio,37670
+workbench/core/artifacts/monitor_core.py,sha256=AIgceuO_YqAmttDdJWcdFcrvE93icfSsTr6WgXWGjYo,37684
 workbench/core/cloud_platform/cloud_meta.py,sha256=-g4-LTC3D0PXb3VfaXdLR1ERijKuHdffeMK_zhD-koQ,8809
 workbench/core/cloud_platform/aws/README.md,sha256=QT5IQXoUHbIA0qQ2wO6_2P2lYjYQFVYuezc22mWY4i8,97
 workbench/core/cloud_platform/aws/aws_account_clamp.py,sha256=OzFknZXKW7VTvnDGGX4BXKoh0i1gQ7yaEBhkLCyHFSs,6310
 workbench/core/cloud_platform/aws/aws_df_store.py,sha256=utRIlTCPwFneHHZ8_Z3Hw3rOJSeryiFA4wBtucxULRQ,15055
 workbench/core/cloud_platform/aws/aws_graph_store.py,sha256=ytYxQTplUmeWbsPmxyZbf6mO9qyTl60ewlJG8MyfyEY,9414
-workbench/core/cloud_platform/aws/aws_meta.py,sha256=ZCKr4cMc0XE9HC0FnLJM1wS85kK8zbzo54OIRN7MiLE,34591
+workbench/core/cloud_platform/aws/aws_meta.py,sha256=xpidYpDydgWmKmJPrNFWbggahDY-nRXzXTRaEA3c5Sc,34587
 workbench/core/cloud_platform/aws/aws_parameter_store.py,sha256=9ekuMOQFHFMIEV68UbHhS_fLB9iqG5Hvu4EV6iamEpk,10400
 workbench/core/cloud_platform/aws/aws_secrets_manager.py,sha256=TUnddp1gX-OwxJ_oO5ONh7OI4Z2HC_6euGkJ-himCCk,8615
-workbench/core/cloud_platform/aws/aws_session.py,sha256=IIGz0ekbNunWzQaeaZzC2-Vl49o4Lv2F35vLtgjMGsQ,6972
+workbench/core/cloud_platform/aws/aws_session.py,sha256=T0Vt8K2MRRznOwxC_MqVgGg2p_CLr0jb3hB5LpwXy_w,6980
 workbench/core/cloud_platform/aws/cache_dataframe.py,sha256=VnObkVqcjg7v4fegrIkXR1j-K2AHTBpSAoriUXDe12A,2314
 workbench/core/cloud_platform/azure/README.md,sha256=ciIXZwjtOPYf9ViquFQxjLKuFwje_hZJHJ2hMQghziI,101
 workbench/core/cloud_platform/gcp/README.md,sha256=MzObe3mWQzjviKD2aXlAV9r_bU4HzTJGapWRsFn6pCU,106
@@ -162,7 +162,7 @@ workbench/model_scripts/xgb_model/generated_model_script.py,sha256=mYl7Wfokbynqr
 workbench/model_scripts/xgb_model/requirements.txt,sha256=jWlGc7HH7vqyukTm38LN4EyDi8jDUPEay4n45z-30uc,104
 workbench/model_scripts/xgb_model/xgb_model.template,sha256=RaUr8X6al5R2IILNKgGUH05Gb4H7AFFG9RE524_VH7Q,17935
 workbench/repl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-workbench/repl/workbench_shell.py,sha256=ms9nVFfKohK8efmiQ2YbOH1OYBRWLgqbByshkcoKDog,22137
+workbench/repl/workbench_shell.py,sha256=eJ3rpYgEwZjhrVVCaJHht2N5BrimN6mbxqHXGrJmwC8,22130
 workbench/resources/open_source_api.key,sha256=3S0OTblsmC0msUPdE_dbBmI83xJNmYscuwLJ57JmuOc,433
 workbench/resources/signature_verify_pub.pem,sha256=V3-u-3_z2PH-805ybkKvzDOBwAbvHxcKn0jLBImEtzM,272
 workbench/scripts/check_double_bond_stereo.py,sha256=p5hnL54Weq77ES0HCELq9JeoM-PyUGkvVSeWYF2dKyo,7776
@@ -193,13 +193,13 @@ workbench/utils/ai_compound_generator.py,sha256=8no4ufP1LJhQfPqaDHvqUYOAh1kzeDVe
 workbench/utils/ai_summary.py,sha256=KwutaozocDxrfErodYpFCDmt1c8zVKf3Jnu4UzBMLJU,4155
 workbench/utils/ai_synth.py,sha256=3mz7NwZeOuNH5ju5n_pikOwfxtVN6LaipLFkAZqoN2U,5607
 workbench/utils/athena_utils.py,sha256=DDyLhJujzh1PfejtGU7ZzOf5hLPOgoXmi4Lrn-_AJzU,4812
-workbench/utils/aws_utils.py,sha256=XckM0vzud7Nx1OxD1GoYGLQxdj1PqeQ43cN66tnrRYI,22002
+workbench/utils/aws_utils.py,sha256=x8c_WxtdSKmBqNg8P_Z6K2m4AsSMEiD_kh2nVaUZ28c,22077
 workbench/utils/bulk_utils.py,sha256=s1lYN2Uk536MNGetekLYL_VL0N34hUjk1FX9BAz3Qu0,1182
 workbench/utils/cache.py,sha256=0R5RXYEz_XHARK3anmQC4VRMawMks_cJ8S4vwC2roAE,5524
 workbench/utils/chem_utils.py,sha256=tLTAvLKTOiYSzbVQF0M8V5-ej36IVgr21CNB2vVJjYQ,56780
-workbench/utils/cloudwatch_handler.py,sha256=dtnkr8tXtTRAASQ60QO0lz3SRA5LEbzsK1VCIqblfKs,5157
+workbench/utils/cloudwatch_handler.py,sha256=t0L280Qa1nMq95dwnf8lB5g8FHrQAyGY5S4JwP3yIa8,5165
 workbench/utils/color_utils.py,sha256=TmDGLK44t975lkfjt_1O-ee02QxrKfke7vPuXb-V-Uo,11779
-workbench/utils/config_manager.py,sha256=Yj43Ta67dn34XdOcKcOvXw38ln6TRiv4DePXjPt2jg8,17641
+workbench/utils/config_manager.py,sha256=SBBmO1RGCQ_Zyh91tDxL1HOm5B0v38ImlLnOsEKzXPU,17649
 workbench/utils/dashboard_metrics.py,sha256=cNFI0GIAjd_IiDzM1oebsJ2QkRZuW068W_66ZC3J100,7398
 workbench/utils/datetime_utils.py,sha256=r3G_KB2euu26lwVbDXYXPJEpJCZwin2Iph7BiBIoybg,4454
 workbench/utils/deprecated_utils.py,sha256=qniHVpDGuwOnhxn65LofDQ_EA2OhSUcZLPxAXtx7FgA,3540
@@ -207,7 +207,6 @@ workbench/utils/df_to_endpoint.py,sha256=bIb1CDko8_BClX5wcQuBbmcGH79n3oHUcb_1jdU
 workbench/utils/ecs_info.py,sha256=Gs9jNb4vcj2pziufIOI4BVIH1J-3XBMtWm1phVh8oRY,2873
 workbench/utils/endpoint_metrics.py,sha256=_4WVU6cLLuV0t_i0PSvhi0EoA5ss5aDFe7ZDpumx2R8,7822
 workbench/utils/endpoint_utils.py,sha256=3-njrhMSAIOaEEiH7qMA9vgD3I7J2S9iUAcqXKx3OBo,7104
-workbench/utils/execution_environment.py,sha256=n8XJa5-fNXYeUA6YFXdk3Z9ZVwIlmakF6dQpwCl5gK0,6920
 workbench/utils/extract_model_artifact.py,sha256=sFwkJd5mfJ1PU37pIHVmUIQS-taIUJdqi3D9-qRmy8g,7870
 workbench/utils/fast_inference.py,sha256=Sm0EV1oPsYYGqiDBVUu3Nj6Ti68JV-UR2S0ZliBDPTk,6148
 workbench/utils/glue_utils.py,sha256=dslfXQcJ4C-mGmsD6LqeK8vsXBez570t3fZBVZLV7HA,2039
@@ -275,9 +274,9 @@ workbench/web_interface/page_views/main_page.py,sha256=X4-KyGTKLAdxR-Zk2niuLJB2Y
 workbench/web_interface/page_views/models_page_view.py,sha256=M0bdC7bAzLyIaE2jviY12FF4abdMFZmg6sFuOY_LaGI,2650
 workbench/web_interface/page_views/page_view.py,sha256=Gh6YnpOGlUejx-bHZAf5pzqoQ1H1R0OSwOpGhOBO06w,455
 workbench/web_interface/page_views/pipelines_page_view.py,sha256=v2pxrIbsHBcYiblfius3JK766NZ7ciD2yPx0t3E5IJo,2656
-workbench-0.8.165.dist-info/licenses/LICENSE,sha256=z4QMMPlLJkZjU8VOKqJkZiQZCEZ--saIU2Z8-p3aVc0,1080
-workbench-0.8.165.dist-info/METADATA,sha256=ET9NQ-NKZIP4pJelWJ3R_SqXS4Nn6C1MhQ5CtZIqu1o,9209
-workbench-0.8.165.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-workbench-0.8.165.dist-info/entry_points.txt,sha256=oZykkheWiiIBjRE8cS5SdcxwmZKSFaQEGwMBjNh-eNM,238
-workbench-0.8.165.dist-info/top_level.txt,sha256=Dhy72zTxaA_o_yRkPZx5zw-fwumnjGaeGf0hBN3jc_w,10
-workbench-0.8.165.dist-info/RECORD,,
+workbench-0.8.166.dist-info/licenses/LICENSE,sha256=z4QMMPlLJkZjU8VOKqJkZiQZCEZ--saIU2Z8-p3aVc0,1080
+workbench-0.8.166.dist-info/METADATA,sha256=a7nuZYqwYcDrQwDLekEr7_Afyxv2IUjxZoD1YwIqzwo,9210
+workbench-0.8.166.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+workbench-0.8.166.dist-info/entry_points.txt,sha256=oZykkheWiiIBjRE8cS5SdcxwmZKSFaQEGwMBjNh-eNM,238
+workbench-0.8.166.dist-info/top_level.txt,sha256=Dhy72zTxaA_o_yRkPZx5zw-fwumnjGaeGf0hBN3jc_w,10
+workbench-0.8.166.dist-info/RECORD,,

workbench/utils/execution_environment.py

@@ -1,211 +0,0 @@
-"""ExecutionEnvironment provides logic/functionality to figure out the current execution environment"""
-
-import os
-import sys
-import logging
-import requests
-from typing import Union
-import boto3
-from datetime import datetime, timezone
-
-# Workbench imports
-from workbench.utils.glue_utils import get_resolved_options
-from workbench.utils.deprecated_utils import deprecated
-
-# Set up the logger
-log = logging.getLogger("workbench")
-
-
-def running_on_glue():
-    """
-    Check if the current execution environment is an AWS Glue job.
-
-    Returns:
-        bool: True if running in AWS Glue environment, False otherwise.
-    """
-    # Check if GLUE_VERSION or GLUE_PYTHON_VERSION is in the environment
-    if "GLUE_VERSION" in os.environ or "GLUE_PYTHON_VERSION" in os.environ:
-        log.info("Running in AWS Glue Environment...")
-        return True
-    else:
-        return False
-
-
-def running_on_lambda():
-    """
-    Check if the current execution environment is an AWS Lambda function.
-
-    Returns:
-        bool: True if running in AWS Lambda environment, False otherwise.
-    """
-    if "AWS_LAMBDA_FUNCTION_NAME" in os.environ:
-        log.info("Running in AWS Lambda Environment...")
-        return True
-    else:
-        return False
-
-
-def running_on_docker() -> bool:
-    """Check if the current environment is running on a Docker container.
-
-    Returns:
-        bool: True if running in a Docker container, False otherwise.
-    """
-    try:
-        # Docker creates a .dockerenv file at the root of the directory tree inside the container.
-        # If this file exists, it is very likely that we are running inside a Docker container.
-        with open("/.dockerenv") as f:
-            return True
-    except FileNotFoundError:
-        pass
-
-    try:
-        # Another method is to check the contents of /proc/self/cgroup which should be different
-        # inside a Docker container.
-        with open("/proc/self/cgroup") as f:
-            if any("docker" in line for line in f):
-                return True
-    except FileNotFoundError:
-        pass
-
-    # Check if we are running on ECS
-    if running_on_ecs():
-        return True
-
-    # Probably not running in a Docker container
-    return False
-
-
-def running_on_ecs() -> bool:
-    """
-    Check if the current environment is running on AWS ECS.
-
-    Returns:
-        bool: True if running on AWS ECS, False otherwise.
-    """
-    indicators = [
-        "ECS_SERVICE_NAME",
-        "ECS_CONTAINER_METADATA_URI",
-        "ECS_CONTAINER_METADATA_URI_V4",
-        "AWS_CONTAINER_CREDENTIALS_RELATIVE_URI",
-        "AWS_EXECUTION_ENV",
-    ]
-    return any(indicator in os.environ for indicator in indicators)
-
-
-def running_as_service() -> bool:
-    """
-    Check if the current environment is running as a service (e.g. Docker, ECS, Glue, Lambda).
-
-    Returns:
-        bool: True if running as a service, False otherwise.
-    """
-    return running_on_docker() or running_on_glue() or running_on_lambda()
-
-
-def _glue_job_from_script_name(args):
-    """Get the Glue Job Name from the script name"""
-    try:
-        script_name = args["scriptLocation"]
-        return os.path.splitext(os.path.basename(script_name))[0]
-    except Exception:
-        return "unknown"
-
-
-def glue_job_name():
-    """Get the Glue Job Name from the environment or script name"""
-    # Define the required argument
-    args = get_resolved_options(sys.argv)
-
-    # Get the job name
-    job_name = args.get("JOB_NAME") or _glue_job_from_script_name(args)
-    return job_name
-
-
-@deprecated(version=0.9)
-def glue_job_run_id(job_name: str, session: boto3.Session) -> Union[str, None]:
-    """Retrieve the Glue Job Run ID closest to the current time for the given job name.
-    Note: This mostly doesn't work, it will grab A glue job id but often not the correct one.
-          For now, I would just skip using this
-    """
-    try:
-        # Set current time in UTC
-        current_time = datetime.now(timezone.utc)
-
-        job_runs = session.client("glue").get_job_runs(JobName=job_name)
-        if job_runs["JobRuns"]:
-            # Find the job run with the StartedOn time closest to the current time
-            closest_job_run = min(job_runs["JobRuns"], key=lambda run: abs(run["StartedOn"] - current_time))
-            job_id = closest_job_run["Id"]
-            return job_id[:9]  # Shorten the Job Run ID to 9 characters
-
-        log.error(f"No runs found for Glue Job '{job_name}', returning None for Job Run ID.")
-        return None
-
-    except session.client("glue").exceptions.EntityNotFoundException:
-        log.error(f"Glue Job '{job_name}' not found, returning None for Job Run ID.")
-        return None
-    except Exception as e:
-        log.error(f"An error occurred while retrieving job run ID: {e}")
-        return None
-
-
-def ecs_job_name():
-    """Get the ECS Job Name from the metadata endpoint or environment variables."""
-    # Attempt to get the job name from ECS metadata
-    ecs_metadata_uri = os.environ.get("ECS_CONTAINER_METADATA_URI_V4")
-
-    if ecs_metadata_uri:
-        try:
-            response = requests.get(f"{ecs_metadata_uri}/task")
-            if response.status_code == 200:
-                metadata = response.json()
-                job_name = metadata.get("Family")  # 'Family' represents the ECS task definition family name
-                if job_name:
-                    return job_name
-        except requests.RequestException as e:
-            # Log the error or handle it as needed
-            log.error(f"Failed to fetch ECS metadata: {e}")
-
-    # Fallback to environment variables if metadata is not available
-    job_name = os.environ.get("ECS_SERVICE_NAME", "unknown")
-    return job_name
-
-
-if __name__ == "__main__":
-    """Test the Execution Environment utilities"""
-
-    # Test running_on_glue
-    assert running_on_glue() is False
-    os.environ["GLUE_VERSION"] = "1.0"
-    assert running_on_glue() is True
-    del os.environ["GLUE_VERSION"]
-
-    # Test running_on_lambda
-    assert running_on_lambda() is False
-    os.environ["AWS_LAMBDA_FUNCTION_NAME"] = "my_lambda_function"
-    assert running_on_lambda() is True
-    del os.environ["AWS_LAMBDA_FUNCTION_NAME"]
-
-    # Test running_on_docker
-    assert running_on_docker() is False
-    os.environ["ECS_CONTAINER_METADATA_URI"] = "http://localhost:8080"
-    assert running_on_docker() is True
-    del os.environ["ECS_CONTAINER_METADATA_URI"]
-
-    # Test running_on_ecs
-    assert running_on_ecs() is False
-    os.environ["ECS_CONTAINER_METADATA_URI"] = "http://localhost:8080"
-    assert running_on_ecs() is True
-    del os.environ["ECS_CONTAINER_METADATA_URI"]
-
-    # Test getting the Glue Job Name
-    print(glue_job_name())
-
-    # Test getting the Glue Job Run ID
-    from workbench.core.cloud_platform.aws.aws_session import AWSSession
-
-    session = AWSSession().boto3_session
-    print(glue_job_run_id("Test_Workbench_Shell", session))
-
-    print("All tests passed!")