Fix or ignore some pytype errors.

PiperOrigin-RevId: 412228893
This commit is contained in:
DeepMind Team
2021-11-25 09:45:10 +00:00
committed by Diego de Las Casas
parent 2c7c401024
commit c6220061a7
+1 -1
View File
@@ -145,7 +145,7 @@ def main(_):
FLAGS.batch_size)
else:
raise ValueError(f'Unknown experiment mode: {FLAGS.experiment_mode}')
config['checkpointing_config']['checkpoint_dir'] = FLAGS.checkpoint_root
config['checkpointing_config']['checkpoint_dir'] = FLAGS.checkpoint_root # pytype: disable=unsupported-operands # dict-kwargs
if FLAGS.worker_mode == 'train':
train_loop(experiment_class, config)