def __attrs_post_init__(self):
if self.defined_as is None and self.type is None:
raise ValueError("Either defined_as or type need to be specified")
# load additional objects
for k in self.postprocessing:
k_observed = k
if k == 'variant_effects':
k = 'kipoi_veff'
if is_installed(k):
# Load the config properly if the plugin is installed
try:
parser = get_model_yaml_parser(k)
self.postprocessing[k_observed] = parser.from_config(
self.postprocessing[k_observed])
object.__setattr__(self, "postprocessing",
self.postprocessing)
except Exception:
logger.warn(
"Unable to parse {} filed in ModelDescription: {}".
format(k_observed, self))
# parse args
self.args = recursive_dict_parse(self.args, 'url',
RemoteFile.from_config)
# parse default_dataloader
if isinstance(self.default_dataloader, dict):
self.default_dataloader = DataLoaderImport.from_config(
self.default_dataloader)
def __attrs_post_init__(self):
# load additional objects
for k in self.postprocessing:
k_observed = k
if k == 'variant_effects':
k = 'kipoi_veff'
if is_installed(k):
# Load the config properly if the plugin is installed
try:
parser = get_model_yaml_parser(k)
self.postprocessing[k_observed] = parser.from_config(
self.postprocessing[k_observed])
object.__setattr__(self, "postprocessing",
self.postprocessing)
except Exception:
logger.warn(
"Unable to parse {} filed in ModelDescription: {}".
format(k_observed, self))
def test_yaml_parsers():
plugin = load_plugin("kipoi_veff")
assert plugin.ModelParser == get_model_yaml_parser("kipoi_veff")
assert plugin.DataloaderParser == get_dataloader_yaml_parser("kipoi_veff")
assert get_model_yaml_parser("kipoi_interpret") is None
assert get_dataloader_yaml_parser("kipoi_interpret") is None
