fix: flaky test, issue 223 (#250)

googleapis · Sep 10, 2021 · 035699f · 035699f
1 parent 0fce7c5
commit 035699f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/google-cloud-automl/samples/snippets/vision_classification_create_dataset.py b/packages/google-cloud-automl/samples/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()