|
17 | 17 | from __future__ import absolute_import |
18 | 18 |
|
19 | 19 | from typing import Iterable, List, Tuple # pylint: disable=unused-import |
20 | | -import itertools |
21 | 20 | import logging |
| 21 | +import itertools |
22 | 22 |
|
23 | 23 | import apache_beam as beam |
24 | 24 | from apache_beam import coders |
25 | 25 | from apache_beam import transforms |
26 | 26 | from apache_beam.io import filebasedsource |
27 | | -from apache_beam.io import range_trackers # pylint: disable=unused-import |
28 | 27 | from apache_beam.io import filesystem |
29 | 28 | from apache_beam.io import filesystems |
30 | 29 | from apache_beam.io import iobase |
| 30 | +from apache_beam.io import range_trackers # pylint: disable=unused-import |
31 | 31 |
|
32 | 32 | from gcp_variant_transforms.beam_io import vcfio |
33 | 33 |
|
34 | 34 |
|
35 | 35 | def _get_file_size(file_name): |
36 | 36 | # 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: |
39 | 39 | 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] |
43 | 43 |
|
44 | 44 | compression_type = filesystem.CompressionTypes.detect_compression_type( |
45 | 45 | file_metadata.path) |
|
0 commit comments