Skip to content

Commit 574bc91

Browse files
committed
Fix test import
1 parent 4ea5a40 commit 574bc91

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

label_studio_ml/examples/deepgram/test_model.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,10 @@
77
# Ensure the Label Studio SDK inside the Deepgram example sees harmless defaults.
88
os.environ.setdefault('LABEL_STUDIO_URL', 'http://localhost')
99
os.environ.setdefault('LABEL_STUDIO_API_KEY', 'test-token')
10-
11-
from label_studio_ml.examples.deepgram import model as deepgram_model # noqa: E402
10+
try:
11+
from label_studio_ml.examples.deepgram import model as deepgram_model # noqa: E402
12+
except ImportError:
13+
from model import DeepgramModel as deepgram_model
1214

1315

1416
@pytest.fixture

0 commit comments

Comments
 (0)