diff --git a/.cloud-build/execute_changed_notebooks_helper.py b/.cloud-build/execute_changed_notebooks_helper.py index 95b60414a..26289f89a 100755 --- a/.cloud-build/execute_changed_notebooks_helper.py +++ b/.cloud-build/execute_changed_notebooks_helper.py @@ -238,7 +238,7 @@ def _get_notebook_python_version(notebook_path: str) -> str: # Look for the python version specification pattern re_match = re.search( - "python version = (\d+\.\d+)", markdown, flags=re.IGNORECASE + r"python version = (\d+\.\d+)", markdown, flags=re.IGNORECASE ) if re_match: # get the version number diff --git a/.cloud-build/test_notebook_vm.txt b/.cloud-build/test_notebook_vm.txt index 0d279a8a4..97706e9b8 100644 --- a/.cloud-build/test_notebook_vm.txt +++ b/.cloud-build/test_notebook_vm.txt @@ -1,5 +1,3 @@ -notebooks/official/vizier/gapic-vizier-multi-objective-optimization.ipynb -notebooks/official/pipelines/lightweight_functions_component_io_kfp.ipynb notebooks/official/ml_metadata/sdk-metric-parameter-tracking-for-locally-trained-models.ipynb notebooks/official/custom/custom-tabular-bq-managed-dataset.ipynb .cloud-build/tests/python_version_test.ipynb diff --git a/notebooks/official/custom/custom-tabular-bq-managed-dataset.ipynb b/notebooks/official/custom/custom-tabular-bq-managed-dataset.ipynb index adac27be5..6e5a63890 100644 --- a/notebooks/official/custom/custom-tabular-bq-managed-dataset.ipynb +++ b/notebooks/official/custom/custom-tabular-bq-managed-dataset.ipynb @@ -409,7 +409,7 @@ "df = bq_client.list_rows(table).to_dataframe()\n", "\n", "# Drop unusable rows\n", - "df = df.replace(to_replace=NA_VALUES, value=np.NaN).dropna()\n", + "df = df.replace(to_replace=NA_VALUES, value=np.nan).dropna()\n", "\n", "# Convert categorical columns to numeric\n", "df[\"island\"], _ = pd.factorize(df[\"island\"])\n",