From 9c95b938770d6ddc9b304d7d185a4aedc605bf71 Mon Sep 17 00:00:00 2001 From: marrrcin Date: Wed, 3 Aug 2022 11:14:35 +0200 Subject: [PATCH] #8 Auto-dataset via custom runner - parameter rename --- kedro_vertexai/vertex_ai/datasets.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/kedro_vertexai/vertex_ai/datasets.py b/kedro_vertexai/vertex_ai/datasets.py index a0188c7..11bd168 100644 --- a/kedro_vertexai/vertex_ai/datasets.py +++ b/kedro_vertexai/vertex_ai/datasets.py @@ -15,10 +15,10 @@ 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 @@ -26,7 +26,7 @@ def __init__( 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()