Make jwt optional import in conftest

microsoft · Jul 10, 2024 · 073ff02 · 073ff02
1 parent b3ee5fd
commit 073ff02
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/promptflow-evals/tests/evals/conftest.py b/src/promptflow-evals/tests/evals/conftest.py
@@ -4,7 +4,6 @@
 from typing import Dict
 from unittest.mock import patch
 
-import jwt
 import pytest
 
 from pytest_mock import MockerFixture
@@ -40,6 +39,7 @@ def is_replay():
 # Import of optional packages
 AZURE_INSTALLED = True
 try:
+    import jwt
     from azure.ai.ml._ml_client import MLClient
 except ImportError:
     AZURE_INSTALLED = False