Fixed the issue #2535 Added user followers and following in class User and added test cases for it (#2536)

* Added user following and user followers in the user class

* Added test cases to check for positive user followers and following

* Update tests/test_hf_api.py

---------

Co-authored-by: Amrit Sundarka <[email protected]>
Co-authored-by: Lucain <[email protected]>
Co-authored-by: Lucain <[email protected]>

Author: 3 people

src/huggingface_hub/hf_api.py

@@ -1405,6 +1405,10 @@ class User:
             Number of upvotes received by the user.
         num_likes (`int`, *optional*):
             Number of likes given by the user.
+        num_following (`int`, *optional*):
+            Number of users this user is following.
+        num_followers (`int`, *optional*):
+            Number of users following this user.
         orgs (list of [`Organization`]):
             List of organizations the user is part of.
     """
@@ -1423,6 +1427,8 @@ class User:
     num_papers: Optional[int] = None
     num_upvotes: Optional[int] = None
     num_likes: Optional[int] = None
+    num_following: Optional[int] = None
+    num_followers: Optional[int] = None
     orgs: List[Organization] = field(default_factory=list)
 
     def __init__(self, **kwargs) -> None:
@@ -1439,6 +1445,8 @@ def __init__(self, **kwargs) -> None:
         self.num_papers = kwargs.pop("numPapers", None)
         self.num_upvotes = kwargs.pop("numUpvotes", None)
         self.num_likes = kwargs.pop("numLikes", None)
+        self.num_following = kwargs.pop("numFollowing", None)
+        self.num_followers = kwargs.pop("numFollowers", None)
         self.user_type = kwargs.pop("type", None)
         self.orgs = [Organization(**org) for org in kwargs.pop("orgs", [])]
 

tests/test_hf_api.py

@@ -4093,6 +4093,8 @@ def test_user_overview(self) -> None:
         assert overview.num_upvotes > 10
         assert len(overview.orgs) > 0
         assert any(org.name == "huggingface" for org in overview.orgs)
+        assert overview.num_following > 300
+        assert overview.num_followers > 1000
 
     def test_organization_members(self) -> None:
         members = self.api.list_organization_members("huggingface")