Disable omegaconf environment variables (#845)

mosaicml · Jan 8, 2024 · 4772ba2 · 4772ba2
1 parent c03ca1a
commit 4772ba2
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/scripts/train/train.py b/scripts/train/train.py
@@ -635,6 +635,11 @@ def main(cfg: DictConfig) -> Trainer:
 
 if __name__ == '__main__':
     yaml_path, args_list = sys.argv[1], sys.argv[2:]
+
+    # Disable resolving environment variables through omegaconf.
+    om.clear_resolver('oc.env')
+
+    # Load yaml and cli arguments.
     with open(yaml_path) as f:
         yaml_cfg = om.load(f)
     cli_cfg = om.from_cli(args_list)