workbench 0.8.230__tar.gz → 0.8.233__tar.gz

{workbench-0.8.230 → workbench-0.8.233}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: workbench
-Version: 0.8.230
+Version: 0.8.233
 Summary: Workbench: A Dashboard and Python API for creating and deploying AWS SageMaker Model Pipelines
 Author-email: SuperCowPowers LLC <support@supercowpowers.com>
 License: MIT License
@@ -60,7 +60,7 @@ Requires-Dist: plotly>=6.0.0; extra == "ui"
 Requires-Dist: dash>=3.0.0; extra == "ui"
 Requires-Dist: dash-bootstrap-components>=1.6.0; extra == "ui"
 Requires-Dist: dash-bootstrap-templates>=1.3.0; extra == "ui"
-Requires-Dist: dash_ag_grid<=32.3.2; extra == "ui"
+Requires-Dist: dash_ag_grid>=33.3.3; extra == "ui"
 Requires-Dist: tabulate>=0.9.0; extra == "ui"
 Requires-Dist: matplotlib>=3.9.2; extra == "ui"
 Provides-Extra: dev

{workbench-0.8.230 → workbench-0.8.233}/applications/aws_dashboard/Dockerfile

@@ -1,5 +1,5 @@
 # Define workbench version once at the top
-ARG WORKBENCH_VERSION=0.8.228
+ARG WORKBENCH_VERSION=0.8.232
 
 # Stage 1: Extract dependencies from workbench[ui]
 FROM python:3.12.5-bookworm AS deps-extractor
@@ -18,12 +18,12 @@ EXPOSE 8000
 # Set PYTHONHASHSEED to ensure consistent hashing across processes
 ENV PYTHONHASHSEED=0
 
-# Install system packages and gunicorn in one layer
+# Install system packages and uvicorn in one layer
 RUN apt-get update && apt-get install -y --no-install-recommends \
-    nginx supervisor libnghttp2-dev && \
+    libnghttp2-dev && \
     apt-get autoremove -y && apt-get clean && \
     rm -rf /var/lib/apt/lists/* && \
-    pip install --no-cache-dir gunicorn
+    pip install --no-cache-dir uvicorn asgiref
 
 # Copy and install dependencies (this layer only changes when deps change)
 COPY --from=deps-extractor /tmp/deps.txt /tmp/deps.txt
@@ -34,12 +34,13 @@ ARG WORKBENCH_VERSION
 RUN pip install --no-cache-dir --no-deps "workbench[ui]==${WORKBENCH_VERSION}"
 
 # Copy app code and configuration (after pip installs to avoid cache invalidation)
-COPY . /app
+COPY app.py /app/
+COPY pages/ /app/pages/
+COPY static/ /app/static/
+COPY assets/ /app/assets/
 ARG WORKBENCH_CONFIG
 COPY $WORKBENCH_CONFIG /app/workbench_config.json
 ENV WORKBENCH_CONFIG=/app/workbench_config.json
 
-# Configure Nginx and Supervisor
-COPY nginx.conf /etc/nginx/sites-available/default
-COPY supervisord.conf /etc/supervisor/conf.d/supervisord.conf
-CMD ["/usr/bin/supervisord"]
+# Run Uvicorn with WSGI-to-ASGI wrapper for Flask/Dash
+CMD ["uvicorn", "app:asgi_app", "--host", "0.0.0.0", "--port", "8000"]

{workbench-0.8.230 → workbench-0.8.233}/applications/aws_dashboard/app.py

@@ -68,9 +68,17 @@ app.index_string = """
 </html>
 """
 
-# Note: The 'server' object is required for running the app with NGINX/uWSGI
+# Note: The 'server' object is required for running the app with WSGI servers
 server = app.server
 
+# ASGI wrapper for Uvicorn (only needed in Docker/production)
+try:
+    from asgiref.wsgi import WsgiToAsgi
+
+    asgi_app = WsgiToAsgi(server)
+except ImportError:
+    asgi_app = None  # Not available when running locally with `python app.py`
+
 # Create the settings menu component
 settings_menu = SettingsMenu()
 settings_menu_id = "settings-menu"

{workbench-0.8.230 → workbench-0.8.233}/applications/aws_dashboard/deploy.sh

@@ -35,7 +35,12 @@ fi
 VERSION_TAG="v${VERSION}_amd64"
 
 echo "📦 Building Docker image version: $VERSION_TAG"
-docker build --build-arg WORKBENCH_CONFIG=$CONFIG -t $IMAGE_NAME:$VERSION_TAG --platform $PLATFORM .
+docker buildx build \
+    --platform $PLATFORM \
+    --build-arg WORKBENCH_CONFIG=$CONFIG \
+    -t $IMAGE_NAME:$VERSION_TAG \
+    --load \
+    .
 
 echo "🔑 Logging in to ECR..."
 aws ecr-public get-login-password --region us-east-1 --profile $AWS_PROFILE | docker login --username AWS --password-stdin public.ecr.aws

{workbench-0.8.230 → workbench-0.8.233}/applications/aws_dashboard/pages/models/callbacks.py

@@ -81,10 +81,10 @@ def setup_plugin_callbacks(plugins, scatter_plot_plugin: ScatterPlot, confusion_
     # Callback to update standard plugins (model_details, shap_plot)
     @callback(
         [Output(component_id, prop) for p in standard_plugins for component_id, prop in p.properties],
-        State("model_details-dropdown", "value"),
         Input("models_table", "selectedRows"),
+        State("model_details-dropdown", "value"),
     )
-    def update_standard_plugin_properties(inference_run, selected_rows):
+    def update_standard_plugin_properties(selected_rows, inference_run):
         """Update properties for standard plugins that take a Model object."""
         if not selected_rows or selected_rows[0] is None:
             raise PreventUpdate
@@ -94,6 +94,7 @@ def setup_plugin_callbacks(plugins, scatter_plot_plugin: ScatterPlot, confusion_
 
         all_props = []
         for p in standard_plugins:
+            # Pass current inference_run to preserve user's dropdown selection
             all_props.extend(p.update_properties(model, inference_run=inference_run))
 
         return all_props

{workbench-0.8.230 → workbench-0.8.233}/aws_setup/workbench_dashboard_full/app.py

@@ -12,7 +12,7 @@ print(f"Account: {aws_account}")
 print(f"Region: {aws_region}")
 
 # When you want a different docker image change this line
-dashboard_image = "public.ecr.aws/m6i5k1r2/workbench_dashboard:v0_8_228_amd64"
+dashboard_image = "public.ecr.aws/m6i5k1r2/workbench_dashboard:v0_8_232_amd64"
 
 # Workbench Configuration
 try:

{workbench-0.8.230 → workbench-0.8.233}/aws_setup/workbench_dashboard_lite/app.py

@@ -12,7 +12,7 @@ print(f"Account: {aws_account}")
 print(f"Region: {aws_region}")
 
 # When you want a different docker image change this line
-dashboard_image = "public.ecr.aws/m6i5k1r2/workbench_dashboard:v0_8_228_amd64"
+dashboard_image = "public.ecr.aws/m6i5k1r2/workbench_dashboard:v0_8_232_amd64"
 
 # Workbench Configuration
 try:

{workbench-0.8.230 → workbench-0.8.233}/aws_setup/workbench_dashboard_lite/workbench_dashboard_lite/workbench_dashboard_stack.py

@@ -105,13 +105,13 @@ class WorkbenchDashboardStack(Stack):
             self,
             "WorkbenchTaskDef",
             task_role=workbench_execution_role,
-            memory_limit_mib=2048,
-            cpu=1024,
+            memory_limit_mib=4096,
+            cpu=2048,
         )
         container = task_definition.add_container(
             "WorkbenchContainer",
             image=ecs.ContainerImage.from_registry(props.dashboard_image),
-            memory_limit_mib=2048,
+            memory_limit_mib=4096,
             environment={
                 "REDIS_HOST": redis_endpoint,
                 "WORKBENCH_BUCKET": props.workbench_bucket,
@@ -167,10 +167,10 @@ class WorkbenchDashboardStack(Stack):
             self,
             "WorkbenchService",
             cluster=cluster,
-            cpu=1024,
+            cpu=2048,
             desired_count=1,
             task_definition=task_definition,
-            memory_limit_mib=2048,
+            memory_limit_mib=4096,
             public_load_balancer=props.public,
             security_groups=[lb_security_group],
             open_listener=props.public,

{workbench-0.8.230 → workbench-0.8.233}/pyproject.toml

@@ -52,7 +52,7 @@ ui = [
     "dash>=3.0.0",
     "dash-bootstrap-components>=1.6.0",
     "dash-bootstrap-templates>=1.3.0",
-    "dash_ag_grid<=32.3.2",
+    "dash_ag_grid>=33.3.3",
     "tabulate>=0.9.0",
     "matplotlib>=3.9.2",
 ]

workbench-0.8.233/scripts/admin/redis_reaper.py

@@ -0,0 +1,97 @@
+import argparse
+import redis
+import heapq
+
+
+def redis_reaper(host, port, size_limit=512000, expire=7, batch_size=1000):
+    """Reap keys from Redis based on size and age.
+
+    Args:
+        host (str): Redis host.
+        port (int): Redis port.
+        size_limit (int): Size limit in bytes (default: 500KB).
+        expire (int): Number of days before a key is considered expired (default: 7 days).
+        batch_size (int): Number of keys to process per pipeline batch (default: 1000).
+    """
+    client = redis.Redis(host=host, port=port)
+    total_keys = client.dbsize()
+    keys_deleted = 0
+    expire_seconds = expire * 86400
+    largest_keys = []  # min-heap for top 5
+    keys_to_delete = []
+
+    # Process keys in batches using pipelines
+    batch = []
+    for key in client.scan_iter(count=batch_size):
+        batch.append(key)
+        if len(batch) >= batch_size:
+            keys_deleted += process_batch(client, batch, size_limit, expire_seconds, largest_keys, keys_to_delete)
+            batch = []
+
+    # Process remaining keys
+    if batch:
+        keys_deleted += process_batch(client, batch, size_limit, expire_seconds, largest_keys, keys_to_delete)
+
+    # Batch delete all marked keys
+    if keys_to_delete:
+        client.delete(*keys_to_delete)
+
+    # Report
+    print(f"\nTotal keys: {total_keys}")
+    print("\nTop 5 largest keys (by size):")
+    for size, key in sorted(largest_keys, reverse=True):
+        print(f"Key: {key}, Size: {size} bytes")
+    print(f"\nTotal keys deleted: {keys_deleted}")
+
+
+def process_batch(client, keys, size_limit, expire_seconds, largest_keys, keys_to_delete):
+    """Process a batch of keys using pipeline for efficiency."""
+    # Pipeline to get memory usage and idle time for all keys
+    pipe = client.pipeline(transaction=False)
+    for key in keys:
+        pipe.memory_usage(key)
+        pipe.object("idletime", key)
+    results = pipe.execute()
+
+    deleted = 0
+    for i, key in enumerate(keys):
+        size = results[i * 2] or 0
+        idle_time = results[i * 2 + 1]
+
+        # Track largest keys
+        heapq.heappush(largest_keys, (size, key))
+        if len(largest_keys) > 5:
+            heapq.heappop(largest_keys)
+
+        # Check deletion criteria
+        if size > size_limit:
+            print(f"Deleting key {key} (size: {size} bytes)")
+            keys_to_delete.append(key)
+            deleted += 1
+        elif idle_time is not None and idle_time > expire_seconds:
+            print(f"Deleting key {key} (age: {idle_time / 86400:.2f} days)")
+            keys_to_delete.append(key)
+            deleted += 1
+        elif b"Crashed" in key:
+            print(f"Deleting key {key} (contains 'Crashed')")
+            keys_to_delete.append(key)
+            deleted += 1
+
+    return deleted
+
+
+if __name__ == "__main__":
+    # Argument parsing
+    parser = argparse.ArgumentParser(description="Redis Reaper: Delete keys based on size and age.")
+    parser.add_argument("--host", type=str, default="localhost", help="Redis host (or host:port)")
+    parser.add_argument("-p", "--port", type=int, default=6379, help="Redis port (default: 6379)")
+    parser.add_argument("--size-limit", type=int, default=1048576, help="Size limit in bytes (default: 1MB)")
+    parser.add_argument("--expire", type=int, default=7, help="Expiration days (default: 7 days)")
+    args = parser.parse_args()
+
+    # Parse host:port if provided
+    host, port = (args.host.split(":") + [args.port])[:2]
+    port = int(port)
+
+    # Run the Redis Reaper
+    redis_reaper(host, port, args.size_limit, args.expire)

{workbench-0.8.230 → workbench-0.8.233}/src/workbench/themes/dark/custom.css

@@ -3,6 +3,7 @@ h1, h2, h3, h4 {
 }
 body {
     color: rgb(180, 180, 180); /* We want the text dim white */
+    background: linear-gradient(90deg, rgb(45, 45, 45) 0%, rgb(35, 35, 35) 100%);
 }
 
 /* Custom CSS to style bold text */
@@ -36,21 +37,38 @@ a:hover {
 
 /* AgGrid custom CSS */
 
-/* There's a one pixel border around the grid that we want to remove */
-.ag-root-wrapper {
-    border: none !important; /* Force removal with !important */
-}
-
-
-/* Box shadow and rounded corners for all AgGrid themes */
+/* AG Grid 33+ uses CSS variables for theming - set them at the theme level */
 [class*="ag-theme-"] {
+    --ag-background-color: rgb(40, 40, 40);
+    --ag-odd-row-background-color: rgb(40, 40, 40);
+    --ag-row-background-color: rgb(50, 50, 50);
+    --ag-selected-row-background-color: rgb(60, 70, 90);
+    --ag-row-hover-color: rgb(55, 55, 65);
+    --ag-header-background-color: rgb(35, 35, 35);
+    --ag-border-color: rgba(80, 80, 80, 0.5);
+    --ag-foreground-color: rgb(180, 180, 180);
+    --ag-header-foreground-color: rgb(220, 220, 220);
+    --ag-wrapper-border-radius: 12px;
+
+    /* Box shadow and rounded corners */
     box-shadow: 2px 2px 6px 5px rgba(0, 0, 0, 0.25);
-    border-radius: 12px; /* Rounded corners */
+    border-radius: 12px;
     border: 0.5px solid rgba(0, 0, 0, 0.5);
     margin: 0;
     padding: 0;
 }
 
+/* Remove border from the grid wrapper */
+.ag-root-wrapper {
+    border: none !important;
+}
+
+/* AG Grid container - remove padding but allow shadow overflow */
+div:has(> [class*="ag-theme-"]) {
+    padding: 0 !important;
+    overflow: visible !important;
+}
+
 /* Apply styling to Workbench containers */
 .workbench-container {
     box-shadow: 2px 2px 6px 5px rgba(0, 0, 0, 0.25);

{workbench-0.8.230 → workbench-0.8.233}/src/workbench/themes/dark/plotly.json

@@ -527,7 +527,7 @@
             "style": "dark"
         },
         "paper_bgcolor": "rgba(0, 0, 0, 0.0)",
-        "plot_bgcolor": "rgba(0, 0, 0, 0.0)",
+        "plot_bgcolor": "rgb(40, 40, 40)",
         "polar": {
             "angularaxis": {
                 "gridcolor": "#506784",

{workbench-0.8.230 → workbench-0.8.233}/src/workbench/themes/light/custom.css

@@ -1,8 +1,8 @@
 /* Light Theme Color Palette - Blue-Purple Blend */
 :root {
     /* Primary colors - blue to purple blend, low saturation */
-    --wb-lightest: rgb(200, 205, 215);        /* Very light blue-lavender - backgrounds */
-    --wb-light: rgb(200, 205, 215);           /* Light periwinkle - alternate rows */
+    --wb-lightest: rgb(215, 218, 225);        /* Light blue-lavender - backgrounds */
+    --wb-light: rgb(210, 213, 220);           /* Slightly lighter - alternate rows */
     --wb-medium: rgb(175, 185, 215);          /* Medium blue-purple - headers */
     --wb-accent: rgb(120, 130, 180);          /* Blue-purple accent - borders */
     --wb-dark: rgb(80, 90, 140);              /* Dark blue-purple - strong accents */
@@ -28,14 +28,13 @@
     --wb-dropdown-focus: rgba(215, 215, 215);
     --wb-dropdown-hover: rgba(215, 215, 215);
 
-    /* Container colors (with alpha) */
-    --wb-container-bg: rgba(215, 215, 215, 0.5);
-    --wb-table-cell-bg: rgba(215, 215, 215, 0.5);
+    /* Container colors */
+    --wb-container-bg: rgb(215, 218, 225);
+    --wb-table-cell-bg: rgb(215, 218, 225);
 
     /* Body gradient colors */
     --wb-body-gradient-start: rgb(180, 195, 215);
-    --wb-body-gradient-mid: rgb(170, 175, 205);
-    --wb-body-gradient-end: rgb(185, 175, 205);
+    --wb-body-gradient-end: rgb(195, 205, 220);
 
     /* Table header gradient colors */
     --wb-header-red-start: rgb(200, 120, 120);
@@ -54,7 +53,7 @@ h1, h2, h3, h4 {
 
 body {
     color: var(--wb-text-primary);
-    background: linear-gradient(135deg, var(--wb-body-gradient-start) 0%, var(--wb-body-gradient-mid) 50%, var(--wb-body-gradient-end) 100%);
+    background: linear-gradient(90deg, var(--wb-body-gradient-start) 0%, var(--wb-body-gradient-end) 100%);
 }
 
 /* Custom CSS to style bold text */
@@ -67,6 +66,16 @@ i, em {
     color: var(--wb-text-italic);
 }
 
+/* Default link color */
+a {
+    color: rgb(40, 80, 200); /* Electric blue for links */
+}
+
+/* Hover effect */
+a:hover {
+    color: rgb(60, 100, 240); /* Brighter blue on hover */
+}
+
 /* Reduce spacing around lists */
 p {
     margin-bottom: 0 !important;
@@ -78,40 +87,19 @@ ul, ol {
 
 /* AgGrid custom CSS */
 
-/* Adjust header background color */
-.ag-header {
+/* AG Grid 33+ uses CSS variables for theming - set them at the theme level */
+[class*="ag-theme-"] {
+    --ag-background-color: var(--wb-lightest);
+    --ag-odd-row-background-color: var(--wb-lightest);
+    --ag-row-background-color: var(--wb-light);
+    --ag-selected-row-background-color: var(--wb-medium);
+    --ag-row-hover-color: var(--wb-light);
     --ag-header-background-color: var(--wb-medium);
-}
-
-/* Adjust cell background and text color */
-.ag-cell {
-    background-color: var(--wb-lightest);
-    color: var(--wb-text-primary);
-}
-
-/* Alternate row colors */
-.ag-row:nth-child(even) .ag-cell {
-    background-color: var(--wb-light);
-}
-
-/* AgGrid header text color */
-.ag-header-cell-text {
-    color: var(--wb-text-primary);
-}
-
-/* Selection color for the entire row */
-.ag-row.ag-row-selected .ag-cell {
-    background-color: var(--wb-medium);
-}
-
-/* There's a one pixel border around the grid that we want to remove */
-.ag-root-wrapper {
-    border: none !important; /* Force removal with !important */
-}
+    --ag-border-color: var(--wb-accent);
+    --ag-foreground-color: var(--wb-text-primary);
+    --ag-header-foreground-color: var(--wb-text-primary);
 
-
-/* Box shadow and rounded corners for all AgGrid themes */
-[class*="ag-theme-"] {
+    /* Box shadow and rounded corners */
     box-shadow: 2px 2px 6px 5px var(--wb-shadow);
     border-radius: 12px;
     border: 0.5px solid var(--wb-accent);
@@ -119,6 +107,17 @@ ul, ol {
     padding: 0;
 }
 
+/* Remove border from the grid wrapper */
+.ag-root-wrapper {
+    border: none !important;
+}
+
+/* AG Grid container - remove padding but allow shadow overflow */
+div:has(> [class*="ag-theme-"]) {
+    padding: 0 !important;
+    overflow: visible !important;
+}
+
 /* Apply styling to Workbench containers */
 .workbench-container {
     background-color: var(--wb-container-bg);

{workbench-0.8.230 → workbench-0.8.233}/src/workbench/themes/midnight_blue/custom.css

@@ -37,48 +37,42 @@ a:hover {
 
 /* AgGrid custom CSS */
 
-/* Adjust header background color */
-.ag-header {
+/* AG Grid 33+ uses CSS variables for theming - set them at the theme level */
+[class*="ag-theme-"] {
+    --ag-background-color: rgb(54, 55, 88);
+    --ag-odd-row-background-color: rgb(54, 55, 88);
+    --ag-row-background-color: rgb(58, 59, 92);
+    --ag-selected-row-background-color: rgb(32, 75, 105);
+    --ag-row-hover-color: rgb(52, 73, 100);
     --ag-header-background-color: rgb(42, 85, 115);
-}
-
-/* Adjust cell background */
-.ag-cell {
-    background-color: rgb(57, 63, 96);
-}
-
-/* Alternate row colors */
-.ag-row:nth-child(even) .ag-cell {
-    background-color: rgb(62, 68, 101);
-}
+    --ag-border-color: rgba(80, 85, 115, 0.5);
+    --ag-foreground-color: rgb(210, 210, 210);
+    --ag-header-foreground-color: rgb(220, 220, 220);
 
-/* Selection color for the entire row */
-.ag-row.ag-row-selected .ag-cell {
-    background-color: rgb(32, 75, 105);
+    /* Box shadow and rounded corners */
+    box-shadow: 2px 2px 6px 5px rgba(0, 0, 0, 0.25);
+    border-radius: 12px;
+    border: 0.5px solid rgba(0, 0, 0, 0.5);
+    margin: 0;
+    padding: 0;
 }
 
-
-
-/* There's a one pixel border around the grid that we want to remove */
+/* Remove border from the grid wrapper */
 .ag-root-wrapper {
-    border: none !important; /* Force removal with !important */
+    border: none !important;
 }
 
-
-/* Box shadow and rounded corners for all AgGrid themes */
-[class*="ag-theme-"] {
-    box-shadow: 2px 2px 6px 5px rgba(0, 0, 0, 0.25);
-    border-radius: 12px; /* Rounded corners */
-    border: 0.5px solid rgba(0, 0, 0, 0.5);
-    margin: 0;
-    padding: 0;
+/* AG Grid container - remove padding but allow shadow overflow */
+div:has(> [class*="ag-theme-"]) {
+    padding: 0 !important;
+    overflow: visible !important;
 }
 
 /* Apply styling to Workbench containers */
 .workbench-container {
     box-shadow: 2px 2px 6px 5px rgba(0, 0, 0, 0.25);
     border-radius: 12px; /* Rounded corners */
-    background-color: rgba(45, 41, 76, 0.5); /* Dark background */
+    background-color: rgb(54, 55, 88); /* Dark background */
 }
 
 /* Apply styling to Workbench Offsets */

{workbench-0.8.230 → workbench-0.8.233}/src/workbench/themes/midnight_blue/plotly.json

@@ -528,7 +528,7 @@
             "style": "dark"
         },
         "paper_bgcolor": "rgba(0, 0, 0, 0.0)",
-        "plot_bgcolor": "rgba(0, 0, 0, 0.0)",
+        "plot_bgcolor": "rgb(54, 55, 88)",
         "polar": {
             "angularaxis": {
                 "gridcolor": "#506784",

{workbench-0.8.230 → workbench-0.8.233}/src/workbench/utils/markdown_utils.py

@@ -1,5 +1,7 @@
 """Markdown Utility/helper methods"""
 
+import math
+
 from workbench.utils.symbols import health_icons
 
 
@@ -229,7 +231,9 @@ def df_to_html_table(df, round_digits: int = 2, margin_bottom: int = 30) -> str:
         for val in row:
             # Format value: integers without decimal, floats rounded
             if isinstance(val, float):
-                if val == int(val):
+                if math.isnan(val):
+                    formatted_val = "NaN"
+                elif val == int(val):
                     formatted_val = int(val)
                 else:
                     formatted_val = round(val, round_digits)

{workbench-0.8.230 → workbench-0.8.233}/src/workbench/web_interface/components/plugins/model_details.py

@@ -3,7 +3,7 @@
 from typing import Union
 
 # Dash Imports
-from dash import html, callback, dcc, Input, Output, State
+from dash import html, callback, dcc, no_update, Input, Output, State
 
 # Workbench Imports
 from workbench.api import ModelType, ParameterStore
@@ -71,7 +71,8 @@ class ModelDetails(PluginInterface):
 
         Args:
             model (CachedModel): An instantiated CachedModel object
-            **kwargs: Additional keyword arguments (unused)
+            **kwargs: Additional keyword arguments
+                - inference_run: Current inference run selection (to preserve user's choice)
 
         Returns:
             list: A list of the updated property values for the plugin
@@ -85,10 +86,17 @@ class ModelDetails(PluginInterface):
 
         # Populate the inference runs dropdown
         inference_runs, default_run = self.get_inference_runs()
-        metrics = self.inference_metrics(default_run)
 
-        # Return the updated property values for the plugin
-        return [header, details, inference_runs, default_run, metrics]
+        # Check if we should preserve the current inference run selection
+        current_inference_run = kwargs.get("inference_run")
+        if current_inference_run and current_inference_run in inference_runs:
+            # Preserve the user's selection - use no_update for dropdown value
+            metrics = self.inference_metrics(current_inference_run)
+            return [header, details, inference_runs, no_update, metrics]
+        else:
+            # New model or invalid selection - use default
+            metrics = self.inference_metrics(default_run)
+            return [header, details, inference_runs, default_run, metrics]
 
     def register_internal_callbacks(self):
         @callback(

{workbench-0.8.230 → workbench-0.8.233}/src/workbench/web_interface/components/plugins/scatter_plot.py

@@ -81,7 +81,7 @@ class ScatterPlot(PluginInterface):
                     id=f"{component_id}-graph",
                     figure=self.display_text("Waiting for Data..."),
                     config={"scrollZoom": True},
-                    style={"height": "600px", "width": "100%"},
+                    style={"height": "500px", "width": "100%"},
                     clear_on_unhover=True,
                 ),
                 # Controls: X, Y, Color, Label Dropdowns, and Regression Line Checkbox

{workbench-0.8.230 → workbench-0.8.233}/src/workbench/web_interface/components/settings_menu.py

@@ -75,8 +75,9 @@ class SettingsMenu:
             style={"display": "flex", "flexDirection": "column", "alignItems": "center", "justifyContent": "center"},
         )
 
-        # Build menu items: Status, License, divider, Themes submenu
+        # Build menu items: Home, Status, License, divider, Themes submenu
         menu_items = [
+            dbc.DropdownMenuItem("Home", href="/"),
             dbc.DropdownMenuItem("Status", href="/status", external_link=True, target="_blank"),
             dbc.DropdownMenuItem("License", href="/license", external_link=True, target="_blank"),
             dbc.DropdownMenuItem(divider=True),

{workbench-0.8.230 → workbench-0.8.233}/src/workbench.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: workbench
-Version: 0.8.230
+Version: 0.8.233
 Summary: Workbench: A Dashboard and Python API for creating and deploying AWS SageMaker Model Pipelines
 Author-email: SuperCowPowers LLC <support@supercowpowers.com>
 License: MIT License
@@ -60,7 +60,7 @@ Requires-Dist: plotly>=6.0.0; extra == "ui"
 Requires-Dist: dash>=3.0.0; extra == "ui"
 Requires-Dist: dash-bootstrap-components>=1.6.0; extra == "ui"
 Requires-Dist: dash-bootstrap-templates>=1.3.0; extra == "ui"
-Requires-Dist: dash_ag_grid<=32.3.2; extra == "ui"
+Requires-Dist: dash_ag_grid>=33.3.3; extra == "ui"
 Requires-Dist: tabulate>=0.9.0; extra == "ui"
 Requires-Dist: matplotlib>=3.9.2; extra == "ui"
 Provides-Extra: dev

{workbench-0.8.230 → workbench-0.8.233}/src/workbench.egg-info/SOURCES.txt

@@ -19,9 +19,7 @@ applications/aws_dashboard/README.md
 applications/aws_dashboard/app.py
 applications/aws_dashboard/dashboard
 applications/aws_dashboard/deploy.sh
-applications/aws_dashboard/nginx.conf
 applications/aws_dashboard/open_source_config.json
-applications/aws_dashboard/supervisord.conf
 applications/aws_dashboard/assets/favicon.ico
 applications/aws_dashboard/pages/data_sources/callbacks.py
 applications/aws_dashboard/pages/data_sources/layout.py