Skip to content

Commit b245c13

Browse files
committed
Reorder import
1 parent 6b428db commit b245c13

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

sdks/python/apache_beam/ml/inference/model_manager_it_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
11
import unittest
22
import apache_beam as beam
33
from apache_beam.ml.inference.base import RunInference
4+
from apache_beam.testing.test_pipeline import TestPipeline
5+
from apache_beam.testing.util import assert_that, equal_to
46

57
# pylint: disable=ungrouped-imports
68
try:
@@ -9,8 +11,6 @@
911
except ImportError as e:
1012
raise unittest.SkipTest(
1113
"HuggingFace model handler dependencies are not installed")
12-
from apache_beam.testing.test_pipeline import TestPipeline
13-
from apache_beam.testing.util import assert_that, equal_to
1414

1515

1616
class HuggingFaceGpuTest(unittest.TestCase):

0 commit comments

Comments
 (0)