diff --git a/sdks/python/apache_beam/io/filesystem_test.py b/sdks/python/apache_beam/io/filesystem_test.py index 38ddf262e649..68e9df59a7bd 100644 --- a/sdks/python/apache_beam/io/filesystem_test.py +++ b/sdks/python/apache_beam/io/filesystem_test.py @@ -19,6 +19,7 @@ """Unit tests for filesystem module.""" import bz2 import gzip +import logging import os import tempfile import unittest diff --git a/sdks/python/apache_beam/io/filesystemio_test.py b/sdks/python/apache_beam/io/filesystemio_test.py index 5609113abcce..41f238361044 100644 --- a/sdks/python/apache_beam/io/filesystemio_test.py +++ b/sdks/python/apache_beam/io/filesystemio_test.py @@ -17,6 +17,7 @@ """Tests for filesystemio.""" import io +import logging import multiprocessing import os import threading diff --git a/sdks/python/apache_beam/io/filesystems_test.py b/sdks/python/apache_beam/io/filesystems_test.py index 34ed8d2e10e0..c084a3cf7bb4 100644 --- a/sdks/python/apache_beam/io/filesystems_test.py +++ b/sdks/python/apache_beam/io/filesystems_test.py @@ -19,6 +19,7 @@ """Unit tests for LocalFileSystem.""" import filecmp +import logging import os import shutil import tempfile diff --git a/sdks/python/apache_beam/io/gcp/gcsfilesystem_test.py b/sdks/python/apache_beam/io/gcp/gcsfilesystem_test.py index d2efcdeed863..e8a827e618a7 100644 --- a/sdks/python/apache_beam/io/gcp/gcsfilesystem_test.py +++ b/sdks/python/apache_beam/io/gcp/gcsfilesystem_test.py @@ -18,6 +18,7 @@ """Unit tests for GCS File System.""" +import logging import unittest import mock diff --git a/sdks/python/apache_beam/io/hadoopfilesystem_test.py b/sdks/python/apache_beam/io/hadoopfilesystem_test.py index 91d19a9047c1..dc6afab58c09 100644 --- a/sdks/python/apache_beam/io/hadoopfilesystem_test.py +++ b/sdks/python/apache_beam/io/hadoopfilesystem_test.py @@ -20,6 +20,7 @@ from __future__ import absolute_import import io +import logging import posixpath import unittest diff --git a/sdks/python/apache_beam/io/localfilesystem_test.py b/sdks/python/apache_beam/io/localfilesystem_test.py index 0e13caa926b7..d4450fd55bb9 100644 --- a/sdks/python/apache_beam/io/localfilesystem_test.py +++ b/sdks/python/apache_beam/io/localfilesystem_test.py @@ -19,6 +19,7 @@ """Unit tests for LocalFileSystem.""" import filecmp +import logging import os import shutil import tempfile diff --git a/sdks/python/apache_beam/options/value_provider_test.py b/sdks/python/apache_beam/options/value_provider_test.py index 810f3be6b96c..cdeb78387b70 100644 --- a/sdks/python/apache_beam/options/value_provider_test.py +++ b/sdks/python/apache_beam/options/value_provider_test.py @@ -16,7 +16,7 @@ # """Unit tests for the ValueProvider class.""" - +import logging import unittest from apache_beam.options.pipeline_options import PipelineOptions diff --git a/sdks/python/apache_beam/transforms/create_test.py b/sdks/python/apache_beam/transforms/create_test.py index 754fb31abcd3..b5d02acc8b11 100644 --- a/sdks/python/apache_beam/transforms/create_test.py +++ b/sdks/python/apache_beam/transforms/create_test.py @@ -16,6 +16,7 @@ # """Unit tests for the Create and _CreateSource classes.""" +import logging import unittest from apache_beam import Create diff --git a/sdks/python/apache_beam/transforms/util_test.py b/sdks/python/apache_beam/transforms/util_test.py index d6199030b78e..18efc44e4d34 100644 --- a/sdks/python/apache_beam/transforms/util_test.py +++ b/sdks/python/apache_beam/transforms/util_test.py @@ -17,6 +17,7 @@ """Unit tests for the transform.util classes.""" +import logging import time import unittest