Fix parse_configuration using "in" with ParseConfigOptions

PiperOrigin-RevId: 368257425

Fix parse_configuration using "in" with ParseConfigOptions
PiperOrigin-RevId: 368257425
782a0299 · Hongkun Yu · A. Unique TensorFlower · 3d9ae6de · 782a0299
Commit 782a0299 authored Apr 13, 2021 by Hongkun Yu Committed by A. Unique TensorFlower Apr 13, 2021
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

official/core/train_utils.py official/core/train_utils.py +3 -0

No files found.
--- a/official/core/train_utils.py
+++ b/official/core/train_utils.py
@@ -234,6 +234,9 @@ class ParseConfigOptions:
  tf_data_service: 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):
  """Parses ExperimentConfig from flags."""