fix: flaky test, issue 223 (#250)

GoogleCloudPlatform · Sep 10, 2021 · 8d52bb3 · 8d52bb3
1 parent 3fce81e
commit 8d52bb3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/automl/snippets/vision_classification_create_dataset.py b/automl/snippets/vision_classification_create_dataset.py
@@ -40,7 +40,7 @@ def create_dataset(project_id, display_name):
     )
 
     # Create a dataset with the dataset metadata in the region.
-    response = client.create_dataset(parent=project_location, dataset=dataset)
+    response = client.create_dataset(parent=project_location, dataset=dataset, timeout=300)
 
     created_dataset = response.result()