fix: Unbreak timeouts for Dataset create.

yinghsienwu · copybara-github · commit 328ebac961f5 · 2023-02-16T15:27:05.000-08:00
PiperOrigin-RevId: 510259344
diff --git a/google/cloud/aiplatform/datasets/dataset.py b/google/cloud/aiplatform/datasets/dataset.py
@@ -351,7 +351,7 @@ def _create_and_import(
 
         _LOGGER.log_create_with_lro(cls, create_dataset_lro)
 
-        created_dataset = create_dataset_lro.result()
+        created_dataset = create_dataset_lro.result(timeout=None)
 
         _LOGGER.log_create_complete(cls, created_dataset, "ds")
 
