We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6b428db commit b245c13Copy full SHA for b245c13
sdks/python/apache_beam/ml/inference/model_manager_it_test.py
@@ -1,6 +1,8 @@
1
import unittest
2
import apache_beam as beam
3
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
6
7
# pylint: disable=ungrouped-imports
8
try:
@@ -9,8 +11,6 @@
9
11
except ImportError as e:
10
12
raise unittest.SkipTest(
13
"HuggingFace model handler dependencies are not installed")
-from apache_beam.testing.test_pipeline import TestPipeline
-from apache_beam.testing.util import assert_that, equal_to
14
15
16
class HuggingFaceGpuTest(unittest.TestCase):
0 commit comments