File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed
sdks/python/apache_beam/internal Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -203,12 +203,12 @@ def dump_session(file_path):
203203
204204 with _pickle_lock , open (file_path , 'wb' ) as file :
205205 coder_reg = typecoders .registry .get_custom_type_coder_tuples ()
206- logicaltype_reg = schemas .LogicalType ._known_logical_types .copy ()
206+ logical_type_reg = schemas .LogicalType ._known_logical_types .copy ()
207207
208208 pickler = cloudpickle .CloudPickler (file )
209209 # TODO(https://github.com/apache/beam/issues/18500) add file system registry
210210 # once implemented
211- pickler .dump ({"coder" : coder_reg , "logicaltype " : logicaltype_reg })
211+ pickler .dump ({"coder" : coder_reg , "logical_type " : logical_type_reg })
212212
213213
214214def load_session (file_path ):
@@ -224,7 +224,7 @@ def load_session(file_path):
224224 typecoders .registry .load_custom_type_coder_tuples (registries ["coder" ])
225225 else :
226226 _LOGGER .warning ('No coder registry found in saved session' )
227- if "logicaltype " in registries :
228- schemas .LogicalType ._known_logical_types .load (registries ["logicaltype " ])
227+ if "logical_type " in registries :
228+ schemas .LogicalType ._known_logical_types .load (registries ["logical_type " ])
229229 else :
230230 _LOGGER .warning ('No logical type registry found in saved session' )
You can’t perform that action at this time.
0 commit comments