workbench 0.8.212__py3-none-any.whl → 0.8.213__py3-none-any.whl

workbench/core/artifacts/endpoint_core.py

@@ -476,8 +476,8 @@ class EndpointCore(Artifact):
                         id_column,
                     )
 
-            # For UQ Models we also capture the uncertainty metrics
-            if model.model_type == ModelType.UQ_REGRESSOR:
+            # Capture uncertainty metrics if prediction_std is available (UQ, ChemProp, etc.)
+            if "prediction_std" in prediction_df.columns:
                 metrics = uq_metrics(prediction_df, primary_target)
                 self.param_store.upsert(f"/workbench/models/{model.name}/inference/{capture_name}", metrics)
 
@@ -525,22 +525,20 @@ class EndpointCore(Artifact):
         fs = FeatureSetCore(model.get_input())
         id_column = fs.id_column
 
-        # For UQ models, get UQ columns from training CV results and compute metrics
-        # Note: XGBoost training now saves all UQ columns (q_*, confidence, prediction_std)
-        additional_columns = []
-        if model_type == ModelType.UQ_REGRESSOR:
-            uq_columns = [col for col in out_of_fold_df.columns if col.startswith("q_") or col == "confidence"]
-            if uq_columns:
-                additional_columns = uq_columns
-                self.log.info(f"UQ columns from training: {', '.join(uq_columns)}")
-                primary_target = targets[0] if isinstance(targets, list) else targets
-                metrics = uq_metrics(out_of_fold_df, primary_target)
-                self.param_store.upsert(f"/workbench/models/{model.name}/inference/full_cross_fold", metrics)
-
         # Normalize targets to a list for iteration
         target_list = targets if isinstance(targets, list) else [targets]
         primary_target = target_list[0]
 
+        # Collect UQ columns (q_*, confidence) for additional tracking
+        additional_columns = [col for col in out_of_fold_df.columns if col.startswith("q_") or col == "confidence"]
+        if additional_columns:
+            self.log.info(f"UQ columns from training: {', '.join(additional_columns)}")
+
+        # Capture uncertainty metrics if prediction_std is available (UQ, ChemProp, etc.)
+        if "prediction_std" in out_of_fold_df.columns:
+            metrics = uq_metrics(out_of_fold_df, primary_target)
+            self.param_store.upsert(f"/workbench/models/{model.name}/inference/full_cross_fold", metrics)
+
         # For single-target models (99% of cases), just save as "full_cross_fold"
         # For multi-target models, save each as cv_{target} plus primary as "full_cross_fold"
         is_multi_target = len(target_list) > 1

{workbench-0.8.212.dist-info → workbench-0.8.213.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: workbench
-Version: 0.8.212
+Version: 0.8.213
 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.212.dist-info → workbench-0.8.213.dist-info}/RECORD

@@ -56,7 +56,7 @@ workbench/core/artifacts/data_capture_core.py,sha256=q8f79rRTYiZ7T4IQRWXl8ZvPpcv
 workbench/core/artifacts/data_source_abstract.py,sha256=5IRCzFVK-17cd4NXPMRfx99vQAmQ0WHE5jcm5RfsVTg,10619
 workbench/core/artifacts/data_source_factory.py,sha256=YL_tA5fsgubbB3dPF6T4tO0rGgz-6oo3ge4i_YXVC-M,2380
 workbench/core/artifacts/df_store_core.py,sha256=AueNr_JvuLLu_ByE7cb3u-isH9u0Q7cMP-UCgCX-Ctg,3536
-workbench/core/artifacts/endpoint_core.py,sha256=VkVVnMn6_bVtBLWNEI2hrfW-D4erxZu_sAOQiWqyJkw,56933
+workbench/core/artifacts/endpoint_core.py,sha256=qEjdbzUMeiZZHyJoGQbjCuFKjCVa89fFISnD5hfC7Yw,56787
 workbench/core/artifacts/feature_set_core.py,sha256=wZy-02WXWmSBet5t8mWXFRdv9O4MtW3hWqJuVv7Kok0,39330
 workbench/core/artifacts/model_core.py,sha256=LDH6wgN1521q7klkVF17t7tmueKH095B42_PlSAileo,52251
 workbench/core/artifacts/monitor_core.py,sha256=M307yz7tEzOEHgv-LmtVy9jKjSbM98fHW3ckmNYrwlU,27897
@@ -302,9 +302,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.212.dist-info/licenses/LICENSE,sha256=RTBoTMeEwTgEhS-n8vgQ-VUo5qig0PWVd8xFPKU6Lck,1080
-workbench-0.8.212.dist-info/METADATA,sha256=JaR0G2USi-9YIvcIHrSvVzhCjNDRrHO7In9UzPQNDLA,10492
-workbench-0.8.212.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-workbench-0.8.212.dist-info/entry_points.txt,sha256=viJ6aXRj63sBIs7avj4kFbCO2J2E7jTCrIk8U1SIc3I,511
-workbench-0.8.212.dist-info/top_level.txt,sha256=Dhy72zTxaA_o_yRkPZx5zw-fwumnjGaeGf0hBN3jc_w,10
-workbench-0.8.212.dist-info/RECORD,,
+workbench-0.8.213.dist-info/licenses/LICENSE,sha256=RTBoTMeEwTgEhS-n8vgQ-VUo5qig0PWVd8xFPKU6Lck,1080
+workbench-0.8.213.dist-info/METADATA,sha256=hKbYNrdgKqm0wqivCTflFzZu0ekaU64UdTw5BBamkJc,10492
+workbench-0.8.213.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+workbench-0.8.213.dist-info/entry_points.txt,sha256=viJ6aXRj63sBIs7avj4kFbCO2J2E7jTCrIk8U1SIc3I,511
+workbench-0.8.213.dist-info/top_level.txt,sha256=Dhy72zTxaA_o_yRkPZx5zw-fwumnjGaeGf0hBN3jc_w,10
+workbench-0.8.213.dist-info/RECORD,,