Skip to content

Commit

Permalink
#8 Auto-dataset via custom runner - parameter rename
Browse files Browse the repository at this point in the history
  • Loading branch information
marrrcin committed Aug 3, 2022
1 parent 2fc7697 commit 9c95b93
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions kedro_vertexai/vertex_ai/datasets.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()
Expand Down

0 comments on commit 9c95b93

Please sign in to comment.