From 951c8a158876ef42e64eff71ef8107d4f4338056 Mon Sep 17 00:00:00 2001 From: shaoting-huang Date: Wed, 20 Nov 2024 15:05:32 +0800 Subject: [PATCH] lint fix Signed-off-by: shaoting-huang --- pymilvus/client/prepare.py | 12 +++++++----- pymilvus/milvus_client/milvus_client.py | 2 +- 2 files changed, 8 insertions(+), 6 deletions(-) diff --git a/pymilvus/client/prepare.py b/pymilvus/client/prepare.py index ae8251289..924b5b0c3 100644 --- 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 index b66efd15b..e7456aa8c 100644 --- 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 )