Skip to content

Commit 0ce465c

Browse files
committed
fix(tests): handle missing torch gracefully in test fixtures
- Wrap model loader import in try/except block - Wrap inference engine import in try/except block - Allow tests to run without ML dependencies installed - Improve test isolation for database-focused tests Enables running database tests without full ML stack.
1 parent 252b00d commit 0ce465c

1 file changed

Lines changed: 12 additions & 6 deletions

File tree

tests/conftest.py

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -328,14 +328,20 @@ def reset_singletons() -> Any:
328328

329329
db.connection._db_manager = None
330330

331-
# Reset model loader
332-
import models.loader
331+
# Reset model loader (skip if torch not installed)
332+
try:
333+
import models.loader
333334

334-
models.loader._model_loader = None
335+
models.loader._model_loader = None
336+
except ImportError:
337+
pass # torch not installed, skip model loader reset
335338

336-
# Reset inference engine
337-
import models.inference
339+
# Reset inference engine (skip if torch not installed)
340+
try:
341+
import models.inference
338342

339-
models.inference._inference_engine = None
343+
models.inference._inference_engine = None
344+
except ImportError:
345+
pass # torch not installed, skip inference engine reset
340346

341347
yield

0 commit comments

Comments
 (0)