OneScience-Group/DiffDock

1__all__ = [2    "CGModel",3    "build_score_model",4    "load_model_args",5    "load_score_model",6    "model_uses_lm_embeddings",7]8 9 10def __getattr__(name):11    if name == "CGModel":12        from .cg_model import CGModel13 14        return CGModel15    if name in {16        "build_score_model",17        "load_model_args",18        "load_score_model",19        "model_uses_lm_embeddings",20    }:21        from .score_wrapper import (22            build_score_model,23            load_model_args,24            load_score_model,25            model_uses_lm_embeddings,26        )27 28        return {29            "build_score_model": build_score_model,30            "load_model_args": load_model_args,31            "load_score_model": load_score_model,32            "model_uses_lm_embeddings": model_uses_lm_embeddings,33        }[name]34    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")35