missing import

mosaicml · Nov 17, 2023 · d13fc5b · d13fc5b
1 parent 8e0e9a5
commit d13fc5b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/fixtures/data.py b/tests/fixtures/data.py
@@ -2,7 +2,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 from pathlib import Path
-from unittest.mock import MagicMock
+from unittest.mock import MagicMock, patch
 
 from composer.utils import dist
 from omegaconf import DictConfig