Skip to content

Commit 7333d35

Browse files
committed
Support kernels in list_liked_repos (#4078)
* Support kernels likes * raise badparameter on ValueError
1 parent ea5051f commit 7333d35

3 files changed

Lines changed: 11 additions & 4 deletions

File tree

src/huggingface_hub/cli/buckets.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -960,6 +960,8 @@ def cp(
960960
disable_progress_bars()
961961
try:
962962
api.copy_files(src, dst) # type: ignore
963+
except ValueError as e:
964+
raise typer.BadParameter(str(e))
963965
finally:
964966
if quiet:
965967
enable_progress_bars()

src/huggingface_hub/hf_api.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1617,6 +1617,8 @@ class UserLikes:
16171617
Total number of likes.
16181618
datasets (`list[str]`):
16191619
List of datasets liked by the user (as repo_ids).
1620+
kernels (`list[str]`):
1621+
List of kernels liked by the user (as repo_ids).
16201622
models (`list[str]`):
16211623
List of models liked by the user (as repo_ids).
16221624
spaces (`list[str]`):
@@ -1629,6 +1631,7 @@ class UserLikes:
16291631

16301632
# User likes
16311633
datasets: list[str]
1634+
kernels: list[str]
16321635
models: list[str]
16331636
spaces: list[str]
16341637

@@ -3024,6 +3027,7 @@ def list_liked_repos(
30243027
return UserLikes(
30253028
user=user,
30263029
total=len(likes),
3030+
kernels=[like["repo"]["name"] for like in likes if like["repo"]["type"] == "kernel"],
30273031
models=[like["repo"]["name"] for like in likes if like["repo"]["type"] == "model"],
30283032
datasets=[like["repo"]["name"] for like in likes if like["repo"]["type"] == "dataset"],
30293033
spaces=[like["repo"]["name"] for like in likes if like["repo"]["type"] == "space"],

tests/test_hf_api.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3186,10 +3186,11 @@ def test_list_repo_likers(self) -> None:
31863186
def test_list_likes_on_production(self) -> None:
31873187
# Test julien-c likes a lot of repos !
31883188
likes = HfApi().list_liked_repos("julien-c")
3189-
self.assertEqual(len(likes.models) + len(likes.datasets) + len(likes.spaces), likes.total)
3190-
self.assertGreater(len(likes.models), 0)
3191-
self.assertGreater(len(likes.datasets), 0)
3192-
self.assertGreater(len(likes.spaces), 0)
3189+
assert len(likes.models) + len(likes.datasets) + len(likes.spaces) + len(likes.kernels) == likes.total
3190+
assert len(likes.models) > 0
3191+
assert len(likes.datasets) > 0
3192+
assert len(likes.spaces) > 0
3193+
assert len(likes.kernels) > 0
31933194

31943195

31953196
class TestSquashHistory(HfApiCommonTest):

0 commit comments

Comments
 (0)