From acdf22ac713725e0fe7dec92b33adeae494ba6a2 Mon Sep 17 00:00:00 2001
From: David Cavazos <dcavazos@google.com>
Date: Mon, 9 Jan 2023 01:18:24 +0000
Subject: [PATCH] enable all tests

---
 people-and-planet-ai/weather-forecasting/e2e_test.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/people-and-planet-ai/weather-forecasting/e2e_test.py b/people-and-planet-ai/weather-forecasting/e2e_test.py
index 7fede9987c64..754f5e824faa 100644
--- a/people-and-planet-ai/weather-forecasting/e2e_test.py
+++ b/people-and-planet-ai/weather-forecasting/e2e_test.py
@@ -106,13 +106,13 @@ def test_weather_forecasting_notebook(
         sections={
             "# 📚 Understand the data": {},
             "# 🗄 Create the dataset": {},
-            # "# ☁️ Create the dataset in Dataflow": {
-            #     "replace": {'--runner="DataflowRunner"': dataflow_dataset_flags},
-            # },
+            "# ☁️ Create the dataset in Dataflow": {
+                "replace": {'--runner="DataflowRunner"': dataflow_dataset_flags},
+            },
             "# 🧠 Train the model": {"variables": {"data_path": data_path_local}},
-            # "# ☁️ Train the model in Vertex AI": {
-            #     "variables": {"data_path": data_path_gcs, "epochs": 2}
-            # },
+            "# ☁️ Train the model in Vertex AI": {
+                "variables": {"data_path": data_path_gcs, "epochs": 2}
+            },
             "# 🔮 Make predictions": {"model_path": MODEL_PATH},  # uses model_local_path
         },
     )