lint fix

Signed-off-by: shaoting-huang <shaoting.huang@zilliz.com>
milvus-io · Nov 20, 2024 · 951c8a1 · 951c8a1
1 parent dcde8cc
commit 951c8a1
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/pymilvus/client/prepare.py b/pymilvus/client/prepare.py
@@ -1470,11 +1470,13 @@ def operate_privilege_v2_request(
         db_name: str,
         collection_name: str,
     ):
-        check_pass_param(role_name=role_name)
-        check_pass_param(privilege=privilege)
-        check_pass_param(operate_privilege_type=operate_privilege_type)
-        check_pass_param(db_name=db_name)
-        check_pass_param(collection_name=collection_name)
+        check_pass_param(
+            role_name=role_name,
+            privilege=privilege,
+            operate_privilege_type=operate_privilege_type,
+            db_name=db_name,
+            collection_name=collection_name,
+        )
         return milvus_types.OperatePrivilegeV2Request(
             role=milvus_types.RoleEntity(name=role_name),
             grantor=milvus_types.GrantorEntity(

diff --git a/pymilvus/milvus_client/milvus_client.py b/pymilvus/milvus_client/milvus_client.py
@@ -1027,7 +1027,7 @@ def revoke_privilege_v2(
         **kwargs,
     ):
         conn = self._get_connection()
-        conn.revoke_privilege(
+        conn.revoke_privilege_v2(
             role_name, privilege, db_name, collection_name, timeout=timeout, **kwargs
         )