Skip to content

Commit

Permalink
Roll back CoderRegistry __init__
Browse files Browse the repository at this point in the history
  • Loading branch information
dpcollins-google committed Nov 11, 2021
1 parent b7bfcf3 commit 5e21f43
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions sdks/python/apache_beam/coders/typecoders.py
Original file line number Diff line number Diff line change
Expand Up @@ -80,12 +80,12 @@ def MakeXyzs(v):

class CoderRegistry(object):
"""A coder registry for typehint/coder associations."""
def __init__(self):
def __init__(self, fallback_coder=None):
self._coders = {} # type: Dict[Any, Type[coders.Coder]]
self.custom_types = [] # type: List[Any]
self.register_standard_coders()
self.register_standard_coders(fallback_coder)

def register_standard_coders(self):
def register_standard_coders(self, fallback_coder):
"""Register coders for all basic and composite types."""
# Coders without subclasses.
self._register_coder_internal(int, coders.VarIntCoder)
Expand All @@ -100,7 +100,7 @@ def register_standard_coders(self):
default_fallback_coders = [
coders.ProtoCoder, coders.ProtoPlusCoder, coders.FastPrimitivesCoder
]
self._fallback_coder = FirstOf(default_fallback_coders)
self._fallback_coder = fallback_coder or FirstOf(default_fallback_coders)

def register_fallback_coder(self, fallback_coder):
self._fallback_coder = FirstOf([fallback_coder, self._fallback_coder])
Expand Down

0 comments on commit 5e21f43

Please sign in to comment.