Update src/distilabel_dataset_generator/utils.py

src/distilabel_dataset_generator/utils.py

@@ -48,17 +48,16 @@ def get_duplicate_button():
 
 
 def list_orgs(token: OAuthToken = None):
-    if token is not None:
-        data = whoami(token)
-        organisations = [
-            entry["entity"]["name"]
-            for entry in data["auth"]["accessToken"]["fineGrained"]["scoped"]
-            if "repo.write" in entry["permissions"]
-        ]
-        organisations.append(data["name"])
-        return list(set(organisations))
-    else:
+    if oauth_token is None:
         return []
+    data = whoami(token.token)
+    organisations = [
+        entry["entity"]["name"]
+        for entry in data["auth"]["accessToken"]["fineGrained"]["scoped"]
+        if "repo.write" in entry["permissions"]
+    ]
+    organisations.append(data["name"])
+    return list(set(organisations))
 
 
 def get_org_dropdown(token: OAuthToken = None):