From 8e1f3b84590d5c8c0bc443e33175f22a8e397139 Mon Sep 17 00:00:00 2001 From: Ilya Matiach Date: Tue, 21 Nov 2023 13:15:20 -0500 Subject: [PATCH] increase timeouts for RAI covid healthcare classification notebook to reduce notebook test failures due to job termination --- ...board-healthcare-covid-classification.ipynb | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/sdk/python/responsible-ai/tabular/responsibleaidashboard-healthcare-covid-classification/responsibleaidashboard-healthcare-covid-classification.ipynb b/sdk/python/responsible-ai/tabular/responsibleaidashboard-healthcare-covid-classification/responsibleaidashboard-healthcare-covid-classification.ipynb index 25f87b40864..21704e591b9 100644 --- a/sdk/python/responsible-ai/tabular/responsibleaidashboard-healthcare-covid-classification/responsibleaidashboard-healthcare-covid-classification.ipynb +++ b/sdk/python/responsible-ai/tabular/responsibleaidashboard-healthcare-covid-classification/responsibleaidashboard-healthcare-covid-classification.ipynb @@ -1619,9 +1619,7 @@ "- Add counterfactual analysis to RAI Insights\n", "- Add error analysis to RAI Insights\n", "- Combine all the RAI insights together\n", - "- Generate a scorecard in PDF format for a summary report on model performance\n", - "\n", - "**Note:** the timeout for the counterfactual generation is longer, since this is a comparatively slow process." + "- Generate a scorecard in PDF format for a summary report on model performance" ] }, { @@ -1659,6 +1657,8 @@ " ]\n", ")\n", "\n", + "timeout = 2400\n", + "\n", "\n", "@dsl.pipeline(\n", " compute=compute_name,\n", @@ -1683,21 +1683,21 @@ " categorical_column_names=json.dumps(categorical_features),\n", " classes=classes_in_target,\n", " )\n", - " create_rai_job.set_limits(timeout=1200)\n", + " create_rai_job.set_limits(timeout=timeout)\n", "\n", " # Add an explanation\n", " explain_job = rai_explanation_component(\n", " comment=\"Explanation for the classification dataset\",\n", " rai_insights_dashboard=create_rai_job.outputs.rai_insights_dashboard,\n", " )\n", - " explain_job.set_limits(timeout=1200)\n", + " explain_job.set_limits(timeout=timeout)\n", "\n", " # Add causal analysis\n", " causal_job = rai_causal_component(\n", " rai_insights_dashboard=create_rai_job.outputs.rai_insights_dashboard,\n", " treatment_features=treatment_features,\n", " )\n", - " causal_job.set_limits(timeout=1200)\n", + " causal_job.set_limits(timeout=timeout)\n", "\n", " # Add counterfactual analysis\n", " counterfactual_job = rai_counterfactual_component(\n", @@ -1705,13 +1705,13 @@ " total_cfs=10,\n", " desired_class=\"opposite\",\n", " )\n", - " counterfactual_job.set_limits(timeout=1200)\n", + " counterfactual_job.set_limits(timeout=timeout)\n", "\n", " # Add error analysis\n", " erroranalysis_job = rai_erroranalysis_component(\n", " rai_insights_dashboard=create_rai_job.outputs.rai_insights_dashboard,\n", " )\n", - " erroranalysis_job.set_limits(timeout=1200)\n", + " erroranalysis_job.set_limits(timeout=timeout)\n", "\n", " # Combine everything\n", " rai_gather_job = rai_gather_component(\n", @@ -1721,7 +1721,7 @@ " insight_3=counterfactual_job.outputs.counterfactual,\n", " insight_4=erroranalysis_job.outputs.error_analysis,\n", " )\n", - " rai_gather_job.set_limits(timeout=1200)\n", + " rai_gather_job.set_limits(timeout=timeout)\n", "\n", " rai_gather_job.outputs.dashboard.mode = \"upload\"\n", " rai_gather_job.outputs.ux_json.mode = \"upload\"\n",