Commit 98747638 authored by haileyschoelkopf's avatar haileyschoelkopf
Browse files

remove omegaconf usage

parent 57b20eef
...@@ -16,7 +16,6 @@ import gc ...@@ -16,7 +16,6 @@ import gc
import torch import torch
import transformers import transformers
from omegaconf import OmegaConf
from jinja2 import BaseLoader, Environment, StrictUndefined from jinja2 import BaseLoader, Environment, StrictUndefined
from itertools import islice from itertools import islice
...@@ -56,7 +55,7 @@ def simple_parse_args_string(args_string): ...@@ -56,7 +55,7 @@ def simple_parse_args_string(args_string):
if not args_string: if not args_string:
return {} return {}
arg_list = args_string.split(",") arg_list = args_string.split(",")
args_dict = OmegaConf.to_object(OmegaConf.from_dotlist(arg_list)) args_dict = {k: v for k, v in [arg.split("=") for arg in arg_list]}
return args_dict return args_dict
......
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