workbench 0.8.231__tar.gz → 0.8.233__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: workbench
-Version: 0.8.231
+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

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

@@ -1,5 +1,5 @@
 # Define workbench version once at the top
-ARG WORKBENCH_VERSION=0.8.230
+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.231 → 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.231 → 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.231 → 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.231 → 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_230_amd64"
+dashboard_image = "public.ecr.aws/m6i5k1r2/workbench_dashboard:v0_8_232_amd64"
 
 # Workbench Configuration
 try:

{workbench-0.8.231 → 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_230_amd64"
+dashboard_image = "public.ecr.aws/m6i5k1r2/workbench_dashboard:v0_8_232_amd64"
 
 # Workbench Configuration
 try:

{workbench-0.8.231 → 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.231 → 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.231 → workbench-0.8.233}/src/workbench.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: workbench
-Version: 0.8.231
+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

{workbench-0.8.231 → 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

workbench-0.8.231/applications/compound_explorer/nginx.conf

@@ -1,10 +0,0 @@
-server {
-    listen 80;
-    server_name localhost;
-
-    location / {
-        proxy_pass http://127.0.0.1:8000;
-        proxy_set_header Host $host;
-        proxy_set_header X-Real-IP $remote_addr;
-    }
-}

workbench-0.8.231/applications/compound_explorer/supervisord.conf

@@ -1,22 +0,0 @@
-[supervisord]
-nodaemon=true
-user=root
-
-[program:gunicorn]
-command=gunicorn app:server --bind 0.0.0.0:8000 --workers 2 --timeout 180
-autostart=true
-autorestart=true
-stdout_logfile=/dev/stdout
-stdout_logfile_maxbytes=0
-stderr_logfile=/dev/stderr
-stderr_logfile_maxbytes=0
-
-[program:nginx]
-command=nginx -g 'daemon off;'
-autostart=true
-autorestart=true
-stdout_logfile=/dev/stdout
-stdout_logfile_maxbytes=0
-stderr_logfile=/dev/stderr
-stderr_logfile_maxbytes=0
-