workbench 0.8.230__tar.gz → 0.8.231__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: workbench
-Version: 0.8.230
+Version: 0.8.231
 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.231}/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.230
 
 # Stage 1: Extract dependencies from workbench[ui]
 FROM python:3.12.5-bookworm AS deps-extractor

{workbench-0.8.230 → workbench-0.8.231}/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_230_amd64"
 
 # Workbench Configuration
 try:

{workbench-0.8.230 → workbench-0.8.231}/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_230_amd64"
 
 # Workbench Configuration
 try:

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: workbench
-Version: 0.8.230
+Version: 0.8.231
 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.231}/src/workbench.egg-info/requires.txt

@@ -38,6 +38,6 @@ plotly>=6.0.0
 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.230/scripts/admin/redis_reaper.py

@@ -1,73 +0,0 @@
-import argparse
-import redis
-import heapq
-
-
-def redis_reaper(host, port, size_limit=512000, expire=7):
-    """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).
-    """
-    # Connect to Redis
-    client = redis.Redis(host=host, port=port)
-    total_keys = client.dbsize()
-    keys_deleted = 0
-
-    # Use a min-heap to track the 5 largest sizes
-    largest_keys = []
-
-    # Iterate through all keys
-    for key in client.scan_iter():
-        # Check the size of the value
-        size = client.memory_usage(key) or 0
-
-        # Track the largest keys
-        heapq.heappush(largest_keys, (size, key))
-        if len(largest_keys) > 5:
-            heapq.heappop(largest_keys)
-
-        if size > size_limit:
-            print(f"Deleting key {key} (size: {size} bytes)")
-            client.delete(key)
-            keys_deleted += 1
-            continue
-
-        # Check if the key has a last modified timestamp
-        key_info = client.object("idletime", key)
-        if key_info is not None:
-            key_age_days = key_info / (60 * 60 * 24)
-            if key_age_days > expire:
-                print(f"Deleting key {key} (age: {key_age_days:.2f} days)")
-                client.delete(key)
-                keys_deleted += 1
-
-        # Check if the key has the word "Crashed" in its name
-        if b"Crashed" in key:
-            print(f"Deleting key {key} (contains 'Crashed')")
-            client.delete(key)
-            keys_deleted += 1
-
-    # 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}")
-
-
-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")
-    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()
-
-    # Run the Redis Reaper
-    redis_reaper(args.host, args.port, args.size_limit, args.expire)