#8 Auto-dataset via custom runner - parameter rename

getindata · Aug 3, 2022 · 9c95b93 · 9c95b93
1 parent 2fc7697
commit 9c95b93
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/kedro_vertexai/vertex_ai/datasets.py b/kedro_vertexai/vertex_ai/datasets.py
@@ -15,18 +15,18 @@ def __init__(
         self,
         storage_root: str,
         dataset_name: str,
-        run_id: str,
+        unique_id: str,
     ):
         self.storage_root = storage_root
-        self.run_id = run_id
+        self.unique_id = unique_id
         self.dataset_name = dataset_name
         self.pickle_protocol = None if version_info[:2] > (3, 8) else 4
 
     @lru_cache()
     def _get_target_path(self):
         return (
             f"gs://{self.storage_root.strip('/')}/"
-            f"{KEDRO_VERTEXAI_BLOB_TEMP_DIR_NAME}/{self.run_id}/{self.dataset_name}.bin"
+            f"{KEDRO_VERTEXAI_BLOB_TEMP_DIR_NAME}/{self.unique_id}/{self.dataset_name}.bin"
         )
 
     @lru_cache()