revert default and improved typing

galaxyproject · Aug 22, 2023 · 1847500 · 1847500
1 parent b606f54
commit 1847500
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/bioblend/galaxy/quotas/__init__.py b/bioblend/galaxy/quotas/__init__.py
@@ -17,6 +17,7 @@
     from bioblend.galaxy import GalaxyInstance
 
 QuotaOperations = Literal["+", "-", "="]
+QuotaDefault = Literal["no", "registered", "unregistered"]
 
 
 class QuotaClient(Client):
@@ -80,7 +81,7 @@ def create_quota(
         description: str,
         amount: str,
         operation: QuotaOperations,
-        default: Optional[Literal["no", "registered", "unregistered"]] = "no",
+        default: Optional[QuotaDefault] = "no",
         in_users: Optional[List[str]] = None,
         in_groups: Optional[List[str]] = None,
     ) -> Dict[str, Any]:
@@ -142,7 +143,7 @@ def update_quota(
         description: Optional[str] = None,
         amount: Optional[str] = None,
         operation: Optional[QuotaOperations] = None,
-        default: Optional[str] = None,
+        default: Optional[QuotaDefault] = "no",
         in_users: Optional[List[str]] = None,
         in_groups: Optional[List[str]] = None,
     ) -> str:
@@ -171,8 +172,8 @@ def update_quota(
         :param default: Whether or not this is a default quota. Valid values
                         are ``no``, ``unregistered``, ``registered``.
                         Calling this method with ``default="no"`` on a
-                        non-default quota will throw an error. Not
-                        passing this parameter will leave the default state as it is.
+                        non-default quota will throw an error. None
+                        will leave the default state as it is.
 
         :type in_users: list of str
         :param in_users: A list of user IDs or user emails.