| from header import * |
| import importlib |
|
|
|
|
| def process_caption(caption): |
| caption = re.sub( |
| r"([\"()*#:;~])", |
| " ", |
| caption.lower(), |
| ) |
| caption = re.sub( |
| r"\s{2,}", |
| " ", |
| caption, |
| ) |
| caption = caption.rstrip("\n") |
| caption = caption.strip(" ") |
|
|
| return caption |
|
|
|
|
| def instantiate_from_config(config): |
| if not "target" in config: |
| if config == '__is_first_stage__': |
| return None |
| elif config == "__is_unconditional__": |
| return None |
| raise KeyError("Expected key `target` to instantiate.") |
| return get_obj_from_str(config["target"])(**config.get("params", dict())) |
|
|
|
|
| def get_obj_from_str(string, reload=False): |
| module, cls = string.rsplit(".", 1) |
| if reload: |
| module_imp = importlib.import_module(module) |
| importlib.reload(module_imp) |
| return getattr(importlib.import_module(module, package=None), cls) |
|
|
