fix: flaky test, issue 223 (#250)

andrewferlitsch · dandhlee · commit d2b0ea4877c7 · 2022-11-17T18:42:20.000-05:00
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()
 

Original file line number	Diff line number	Diff line change
`@@ -40,7 +40,7 @@ def create_dataset(project_id, display_name):`
`40`	`40`	`)`
`41`	`41`
`42`	`42`	`# Create a dataset with the dataset metadata in the region.`
`43`		`- response = client.create_dataset(parent=project_location, dataset=dataset)`
	`43`	`+ response = client.create_dataset(parent=project_location, dataset=dataset, timeout=300)`
`44`	`44`
`45`	`45`	`created_dataset = response.result()`
`46`	`46`