From ee6258ef5abfbf110067497975c509033e547a97 Mon Sep 17 00:00:00 2001 From: Reeba Qureshi Date: Fri, 20 Sep 2024 00:36:57 +0530 Subject: [PATCH] fixing lint failures --- sdks/python/apache_beam/yaml/yaml_enrichment.py | 8 +++++--- sdks/python/apache_beam/yaml/yaml_enrichment_test.py | 6 ++++-- 2 files changed, 9 insertions(+), 5 deletions(-) diff --git a/sdks/python/apache_beam/yaml/yaml_enrichment.py b/sdks/python/apache_beam/yaml/yaml_enrichment.py index 77428bbd59f5..4ec8a5a786d3 100644 --- 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 index 9cd28995dfe4..e26d6140af23 100644 --- 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