fixing lint failures

apache · Oct 3, 2024 · ee6258e · ee6258e
1 parent e20166d
commit ee6258e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/sdks/python/apache_beam/yaml/yaml_enrichment.py b/sdks/python/apache_beam/yaml/yaml_enrichment.py
@@ -15,14 +15,16 @@
 # limitations under the License.
 #
 
-from typing import Any, Dict
+from typing import Any
+from typing import Dict
+from typing import Optional
+
 import apache_beam as beam
+from apache_beam.transforms.enrichment import Enrichment
 from apache_beam.transforms.enrichment_handlers.bigquery import BigQueryEnrichmentHandler
 from apache_beam.transforms.enrichment_handlers.bigtable import BigTableEnrichmentHandler
 from apache_beam.transforms.enrichment_handlers.feast_feature_store import FeastFeatureStoreEnrichmentHandler
 from apache_beam.transforms.enrichment_handlers.vertex_ai_feature_store import VertexAIFeatureStoreEnrichmentHandler
-from apache_beam.transforms.enrichment import Enrichment
-from typing import Optional
 
 
 @beam.ptransform.ptransform_fn

diff --git a/sdks/python/apache_beam/yaml/yaml_enrichment_test.py b/sdks/python/apache_beam/yaml/yaml_enrichment_test.py
@@ -15,13 +15,15 @@
 # limitations under the License.
 #
 
-import unittest
 import logging
+import unittest
+
 import mock
+
 import apache_beam as beam
+from apache_beam import Row
 from apache_beam.testing.util import assert_that
 from apache_beam.testing.util import equal_to
-from apache_beam import Row
 from apache_beam.yaml.yaml_transform import YamlTransform