diff --git a/tests/test_hf_conversion_script.py b/tests/test_hf_conversion_script.py index a2de0ea854..d5ee86a406 100644 --- a/tests/test_hf_conversion_script.py +++ b/tests/test_hf_conversion_script.py @@ -5,7 +5,7 @@ import os import pathlib import sys -from unittest.mock import MagicMock, patch, ANY +from unittest.mock import ANY, MagicMock, patch from composer import Trainer from composer.loggers import MLFlowLogger