model/package_config.py

def get_package_data():
    package = "onescience.models.nequip"
    data = {
        package: [
            "LICENSE",
            "NOTICE",
            "SOURCE.md",
            "AUTHORS.md",
            "**/*.yaml",
            "**/*.json",
            "**/*.pt",
        ],
    }
    return data