Skip to content

Commit 68c8dad

Browse files
committed
Make final fixes to PR
1 parent 4ccd72d commit 68c8dad

File tree

2 files changed

+10
-10
lines changed

2 files changed

+10
-10
lines changed

gcp_variant_transforms/beam_io/vcf_file_size_io.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -17,29 +17,29 @@
1717
from __future__ import absolute_import
1818

1919
from typing import Iterable, List, Tuple # pylint: disable=unused-import
20-
import itertools
2120
import logging
21+
import itertools
2222

2323
import apache_beam as beam
2424
from apache_beam import coders
2525
from apache_beam import transforms
2626
from apache_beam.io import filebasedsource
27-
from apache_beam.io import range_trackers # pylint: disable=unused-import
2827
from apache_beam.io import filesystem
2928
from apache_beam.io import filesystems
3029
from apache_beam.io import iobase
30+
from apache_beam.io import range_trackers # pylint: disable=unused-import
3131

3232
from gcp_variant_transforms.beam_io import vcfio
3333

3434

3535
def _get_file_size(file_name):
3636
# type: (str) -> List[FileSizeInfo]
37-
match_result = filesystems.FileSystems.match([file_name])[0]
38-
if len(match_result.metadata_list) != 1:
37+
matched_files = filesystems.FileSystems.match([file_name])[0].metadata_list
38+
if len(matched_files) != 1:
3939
raise IOError("File name {} did not correspond to exactly 1 result. "
40-
"Instead, got {}.".format(file_name,
41-
len(match_result.metadata_list)))
42-
file_metadata = match_result.metadata_list[0]
40+
"Instead, got {} matches.".format(file_name,
41+
len(matched_files)))
42+
file_metadata = matched_files[0]
4343

4444
compression_type = filesystem.CompressionTypes.detect_compression_type(
4545
file_metadata.path)

gcp_variant_transforms/beam_io/vcfio.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -24,14 +24,14 @@
2424
import enum
2525

2626
import apache_beam as beam
27+
from apache_beam import transforms
2728
from apache_beam.coders import coders
2829
from apache_beam.io import filebasedsource
30+
from apache_beam.io import filesystem
2931
from apache_beam.io import filesystems
32+
from apache_beam.io import iobase
3033
from apache_beam.io import range_trackers # pylint: disable=unused-import
3134
from apache_beam.io import textio
32-
from apache_beam.io import filesystem
33-
from apache_beam.io import iobase
34-
from apache_beam import transforms
3535

3636
from gcp_variant_transforms.beam_io import vcf_parser
3737

0 commit comments

Comments
 (0)