Skip to content

Commit 0b83e94

Browse files
authored
Handle directories for dataset/version creation. (#255)
Update kagglehub SDK to support dataset upload with sub-dirs. PRESUBMIT=[kaggleazure pr](Kaggle/kaggleazure#37081) BUGS=439905909 **Tested** Tested locally with - Dataset creation: [new creation of dataset](https://screenshot.googleplex.com/A5tQsZTLSX2ZzLc) - Dataset Version creation: [update dataset to v2](https://screenshot.googleplex.com/BagrxgyzDsURTQB)
1 parent 7cd1dc1 commit 0b83e94

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/kagglehub/datasets_helpers.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ def _create_dataset(dataset_handle: DatasetHandle, files_and_directories: Upload
1414
"ownerSlug": dataset_handle.owner,
1515
"title": dataset_handle.dataset,
1616
"files": [{"token": file_token} for file_token in files_and_directories.files],
17+
"directories": files_and_directories.serialize()["directories"],
1718
"isPrivate": True,
1819
}
1920

@@ -30,6 +31,7 @@ def _create_dataset_version(
3031
data = {
3132
"versionNotes": version_notes,
3233
"files": [{"token": file_token} for file_token in files_and_directories.files],
34+
"directories": files_and_directories.serialize()["directories"],
3335
}
3436
api_client = KaggleApiV1Client()
3537
api_client.post(f"/datasets/create/version/{dataset_handle.owner}/{dataset_handle.dataset}", data)

0 commit comments

Comments
 (0)