fix lint

Signed-off-by: Justin Yu <justinvyu@anyscale.com>
ray-project · Dec 28, 2023 · 8026584 · 8026584
1 parent ec3c988
commit 8026584
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/xgboost_ray/tests/test_tune.py b/xgboost_ray/tests/test_tune.py
@@ -2,7 +2,7 @@
 import shutil
 import tempfile
 import unittest
-from unittest.mock import patch, MagicMock
+from unittest.mock import MagicMock, patch
 
 import numpy as np
 import ray
@@ -83,7 +83,9 @@ def testNumIters(self):
         # TODO(justinvyu): Remove this once the xgboost integration
         # has been updated on the Ray side.
         try:
-            callback = TuneReportCheckpointCallback(frequency=1, checkpoint_at_end=False)
+            callback = TuneReportCheckpointCallback(
+                frequency=1, checkpoint_at_end=False
+            )
         except TypeError:
             callback = TuneReportCheckpointCallback(frequency=1)