From f3288706962298e7306e7d5d9cb0cd13a2ac6d41 Mon Sep 17 00:00:00 2001 From: Pablo Estrada Date: Fri, 20 Nov 2020 08:48:41 -0800 Subject: [PATCH] fixup --- sdks/python/apache_beam/io/gcp/bigquery_read_internal.py | 2 +- sdks/python/apache_beam/io/gcp/bigquery_read_it_test.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/sdks/python/apache_beam/io/gcp/bigquery_read_internal.py b/sdks/python/apache_beam/io/gcp/bigquery_read_internal.py index aef7a871a748a..f1ad38112fc1a 100644 --- a/sdks/python/apache_beam/io/gcp/bigquery_read_internal.py +++ b/sdks/python/apache_beam/io/gcp/bigquery_read_internal.py @@ -26,11 +26,11 @@ import logging import random import uuid +from typing import TYPE_CHECKING from typing import Dict from typing import Iterable from typing import Optional from typing import Union -from typing import TYPE_CHECKING import apache_beam as beam from apache_beam.coders import coders diff --git a/sdks/python/apache_beam/io/gcp/bigquery_read_it_test.py b/sdks/python/apache_beam/io/gcp/bigquery_read_it_test.py index 5df234cc12ea2..4586f046f79bb 100644 --- a/sdks/python/apache_beam/io/gcp/bigquery_read_it_test.py +++ b/sdks/python/apache_beam/io/gcp/bigquery_read_it_test.py @@ -383,6 +383,7 @@ def create_bq_schema(cls, with_extra=False): @skip(['PortableRunner', 'FlinkRunner']) @attr('IT') def test_read_queries(self): + # TODO(BEAM-11311): Remove experiment when tests run on r_v2. args = self.args + ["--experiments=use_runner_v2"] with beam.Pipeline(argv=args) as p: result = (