diff --git a/tensorboard/plugins/core/core_plugin.py b/tensorboard/plugins/core/core_plugin.py index b537b87028..c940e48f98 100644 --- a/tensorboard/plugins/core/core_plugin.py +++ b/tensorboard/plugins/core/core_plugin.py @@ -302,8 +302,9 @@ def define_flags(self, parser): parser.add_argument( '--purge_orphaned_data', metavar='BOOL', - type=bool, - nargs=1, + # Custom str-to-bool converter since regular bool() doesn't work. + type=lambda v: {'true': True, 'false': False}.get(v.lower(), v), + choices=[True, False], default=True, help='''\ Whether to purge data that may have been orphaned due to TensorBoard