enable all tests

GoogleCloudPlatform · Jan 9, 2023 · acdf22a · acdf22a
1 parent 77e32fc
commit acdf22a
Showing 1 changed file with 6 additions and 6 deletions.
diff --git 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
         },
     )