Skip to content
This repository was archived by the owner on Jun 30, 2022. It is now read-only.

Commit 0c4d8c3

Browse files
sbilacaaltay
authored andcommitted
Add utility function to check compression type validity.
----Release Notes---- [] ------------- Created by MOE: https://github.com/google/moe MOE_MIGRATED_REVID=122669970
1 parent 13a733d commit 0c4d8c3

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

google/cloud/dataflow/io/fileio.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -307,6 +307,11 @@ class CompressionTypes(object):
307307
NO_COMPRESSION = _CompressionType(1) # No compression.
308308
DEFLATE = _CompressionType(2) # 'Deflate' ie gzip compression.
309309

310+
@staticmethod
311+
def valid_compression_type(compression_type):
312+
"""Returns true for valid compression types, false otherwise."""
313+
return isinstance(compression_type, _CompressionType)
314+
310315

311316
class FileSink(iobase.Sink):
312317
"""A sink to a GCS or local files.
@@ -445,7 +450,7 @@ def __init__(self,
445450
Returns:
446451
A PureTextFileSink object usable for writing.
447452
"""
448-
if not isinstance(compression_type, _CompressionType):
453+
if not CompressionTypes.valid_compression_type(compression_type):
449454
raise TypeError('compression_type must be CompressionType object but '
450455
'was %s' % type(compression_type))
451456

0 commit comments

Comments
 (0)