Skip to content

GitLab

Commit 782a0299 authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Fix parse_configuration using "in" with ParseConfigOptions 

PiperOrigin-RevId: 368257425
parent 3d9ae6de
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
official/core/train_utils.py
View file @ 782a0299
...@@ -234,6 +234,9 @@ class ParseConfigOptions: ...@@ -234,6 +234,9 @@ class ParseConfigOptions:
tf_data_service: str = '' tf_data_service: str = ''
params_override: str = '' params_override: str = ''
def __contains__(self, name):
return name in dataclasses.asdict(self)
def parse_configuration(flags_obj, lock_return=True, print_return=True): def parse_configuration(flags_obj, lock_return=True, print_return=True):
"""Parses ExperimentConfig from flags.""" """Parses ExperimentConfig from flags."""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment