1 parent d21ac36 commit 9fc2f95Copy full SHA for 9fc2f95
1 file changed
tests/hooks/test_mlflow_tracking.py
@@ -12,7 +12,9 @@
12
# Ensure a mock 'mlflow' module exists so the example can be imported
13
# without mlflow actually installed.
14
if "mlflow" not in sys.modules:
15
- sys.modules["mlflow"] = MagicMock()
+ _mock = MagicMock()
16
+ _mock.__spec__ = None
17
+ sys.modules["mlflow"] = _mock
18
19
from inspect_ai.hooks._hooks import (
20
ModelUsageData,
0 commit comments