<filename>attacut/artifacts/__init__.py
import os

from attacut import logger

log = logger.get_logger(__name__)

artifact_dir = os.path.dirname(__file__)


def get_path(name: str) -> str:
    if name in ["attacut-c", "attacut-sc"]:
        return f"{artifact_dir}/{name}"
    else:
        # if name isn't in the list, then it's a custom model
        log.info("model_path: %s" % name)
        return name

<filename>attacut/artifacts/__init__.py
import os

from attacut import logger

log = logger.get_logger(__name__)

artifact_dir = os.path.dirname(__file__)


def get_path(name: str) -> str:
    if name in ["attacut-c", "attacut-sc"]:
        return f"{artifact_dir}/{name}"
    else:
        # if name isn't in the list, then it's a custom model
        log.info("model_path: %s" % name)
        return name

# if name isn't in the list, then it's a custom model