diff --git a/ydb/iam/auth.py b/ydb/iam/auth.py index 5fd179d2..c4096c09 100644 --- a/ydb/iam/auth.py +++ b/ydb/iam/auth.py @@ -21,8 +21,13 @@ from ydb.public.api.client.yc_public.iam import iam_token_service_pb2_grpc from ydb.public.api.client.yc_public.iam import iam_token_service_pb2 except ImportError: - iam_token_service_pb2_grpc = None - iam_token_service_pb2 = None + try: + # This attempt is to enable the IAM auth inside the YDB repository on Arcadia + from contrib.ydb.public.api.client.yc_public.iam import iam_token_service_pb2_grpc + from contrib.ydb.public.api.client.yc_public.iam import iam_token_service_pb2 + except ImportError: + iam_token_service_pb2_grpc = None + iam_token_service_pb2 = None try: import requests