Merge branch 'main' into fix_is_testing

notebooks/official/automl/sdk_automl_tabular_forecasting_batch.ipynb

@@ -184,7 +184,7 @@
       },
       "outputs": [],
       "source": [
-        "if os.environ[\"IS_TESTING\"]:\n",
+        "if os.getenv(\"IS_TESTING\"):\n",
         "    ! pip3 install --upgrade tensorflow $USER_FLAG"
       ]
     },

notebooks/official/explainable_ai/sdk_custom_tabular_regression_batch_explain.ipynb

@@ -1544,7 +1544,7 @@
       },
       "outputs": [],
       "source": [
-        "if not os.environ[\"IS_TESTING\"]:\n",
+        "if not os.getenv(\"IS_TESTING\"):\n",
         "    batch_predict_job.wait()"
       ]
     },
@@ -1572,7 +1572,7 @@
       },
       "outputs": [],
       "source": [
-        "if not os.environ[\"IS_TESTING\"]:\n",
+        "if not os.getenv(\"IS_TESTING\"):\n",
         "    import tensorflow as tf\n",
         "\n",
         "    bp_iter_outputs = batch_predict_job.iter_outputs()\n",

notebooks/official/migration/UJ1 Vertex SDK AutoML Image Classification.ipynb

@@ -169,7 +169,7 @@
       },
       "outputs": [],
       "source": [
-        "if os.environ[\"IS_TESTING\"]:\n",
+        "if os.getenv(\"IS_TESTING\"):\n",
         "    ! pip3 install --upgrade tensorflow $USER_FLAG"
       ]
     },

notebooks/official/migration/UJ10 Vertex SDK Custom Scikit-Learn with pre-built training container.ipynb

@@ -171,7 +171,7 @@
       },
       "outputs": [],
       "source": [
-        "if os.environ[\"IS_TESTING\"]:\n",
+        "if os.getenv(\"IS_TESTING\"):\n",
         "    ! pip3 install --upgrade tensorflow $USER_FLAG"
       ]
     },

notebooks/official/migration/UJ14 Vertex SDK AutoML Video Classification.ipynb

@@ -169,7 +169,7 @@
       },
       "outputs": [],
       "source": [
-        "if os.environ[\"IS_TESTING\"]:\n",
+        "if os.getenv(\"IS_TESTING\"):\n",
         "    ! pip3 install --upgrade tensorflow $USER_FLAG"
       ]
     },