Skip to content

Commit ff7afe7

Browse files
committed
Add paper_dataset_list method.
1 parent bfffedc commit ff7afe7

File tree

1 file changed

+21
-0
lines changed

1 file changed

+21
-0
lines changed

paperswithcode/client.py

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -139,6 +139,27 @@ def paper_get(self, paper_id: str) -> Paper:
139139
"""
140140
return Paper(**self.http.get(f"/papers/{paper_id}/"))
141141

142+
@handler
143+
def paper_dataset_list(
144+
self, paper_id: str, page: int = 1, items_per_page: int = 50
145+
) -> Repositories:
146+
"""Return a list of datasets mentioned in the paper..
147+
148+
Args:
149+
paper_id (str): ID of the paper.
150+
page (int): Desired page.
151+
items_per_page (int): Desired number of items per page.
152+
Default: 50.
153+
154+
Returns:
155+
Datasets: Datasets object.
156+
"""
157+
params = self.__params(page, items_per_page)
158+
return self.__page(
159+
self.http.get(f"/papers/{paper_id}/datasets/", params=params),
160+
Datasets,
161+
)
162+
142163
@handler
143164
def paper_repository_list(
144165
self, paper_id: str, page: int = 1, items_per_page: int = 50

0 commit comments

Comments
 (0)