Skip to content

Commit d195d71

Browse files
committed
fix naming
1 parent 0091b24 commit d195d71

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

sdks/python/apache_beam/internal/cloudpickle_pickler.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff 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

214214
def 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')

0 commit comments

Comments
 (0)