setup.py

import platform
from setuptools import setup, find_packages

if platform.python_version_tuple()[:2] != ("3", "11"):
    raise RuntimeError("Python version 3.11 required")

setup(
    name="distilvit",
    version="0.1",
    packages=find_packages(),
    entry_points={
        "console_scripts": [
            "train=src.model.train:main",  # "main" is a function in "train_model.py"
        ],
    },
)