diff --git a/tests/test_trainer_utils.py b/tests/test_trainer_utils.py index beed009a5..2fc5b3a2e 100644 --- a/tests/test_trainer_utils.py +++ b/tests/test_trainer_utils.py @@ -1,4 +1,5 @@ """Module containing tests for trainer utility functions.""" + import unittest from functools import partial