Skip to content

Commit 4edc26c

Browse files
authored
Fix dataset version spec (#862)
Fix #860
1 parent de70d65 commit 4edc26c

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

src/kaggle/api/kaggle_api_extended.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1994,7 +1994,7 @@ def dataset_list_files(self, dataset, page_token=None, page_size=20):
19941994
request = ApiListDatasetFilesRequest()
19951995
request.owner_slug = owner_slug
19961996
request.dataset_slug = dataset_slug
1997-
request.dataset_version_number = dataset_version_number
1997+
request.dataset_version_number = int(dataset_version_number) if dataset_version_number else None
19981998
request.page_token = page_token
19991999
request.page_size = page_size
20002000
response = kaggle.datasets.dataset_api_client.list_dataset_files(request)
@@ -2096,7 +2096,7 @@ def dataset_download_file(self, dataset, file_name, path=None, force=False, quie
20962096
request = ApiDownloadDatasetRequest()
20972097
request.owner_slug = owner_slug
20982098
request.dataset_slug = dataset_slug
2099-
request.dataset_version_number = dataset_version_number
2099+
request.dataset_version_number = int(dataset_version_number) if dataset_version_number else None
21002100
request.file_name = file_name
21012101
response = kaggle.datasets.dataset_api_client.download_dataset(request)
21022102
url = response.request.url
@@ -2133,7 +2133,7 @@ def dataset_download_files(self, dataset, path=None, force=False, quiet=True, un
21332133
request = ApiDownloadDatasetRequest()
21342134
request.owner_slug = owner_slug
21352135
request.dataset_slug = dataset_slug
2136-
request.dataset_version_number = dataset_version_number
2136+
request.dataset_version_number = int(dataset_version_number) if dataset_version_number else None
21372137
response = kaggle.datasets.dataset_api_client.download_dataset(request)
21382138

21392139
outfile = os.path.join(effective_path, dataset_slug + ".zip")

tests/unit_tests.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -490,7 +490,7 @@ def test_dataset_g_download_files(self):
490490
self.test_dataset_a_list()
491491
ds = ["a", "b"]
492492
try:
493-
api.dataset_download_files(self.dataset)
493+
api.dataset_download_files(self.dataset + "/1")
494494
ds = self.dataset.split("/")
495495
self.assertTrue(os.path.exists(f"{ds[1]}.zip"))
496496
except ApiException as e:

0 commit comments

Comments
 (0)