Commit 98747638 authored by haileyschoelkopf's avatar haileyschoelkopf
Browse files

remove omegaconf usage

parent 57b20eef
......@@ -16,7 +16,6 @@ import gc
import torch
import transformers
from omegaconf import OmegaConf
from jinja2 import BaseLoader, Environment, StrictUndefined
from itertools import islice
......@@ -56,7 +55,7 @@ def simple_parse_args_string(args_string):
if not args_string:
return {}
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
......
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