resolve imports.

a-r-r-o-w · Jan 27, 2025 · 3609fdf · 3609fdf
1 parent c432f39
commit 3609fdf
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/trainers/cogvideox/test_cogvideox.py b/tests/trainers/cogvideox/test_cogvideox.py
@@ -1,16 +1,14 @@
 import sys
+import unittest
 from pathlib import Path
+from typing import Tuple
 
 
 current_file = Path(__file__).resolve()
 root_dir = current_file.parents[3]
 sys.path.append(str(root_dir))
 
-import unittest  # noqa
-from typing import Tuple  # noqa
-
 from finetrainers import Args  # noqa
-
 from ..test_trainers_common import TrainerTestMixin  # noqa