setup.py

from pybind11.setup_helpers import Pybind11Extension, build_ext
from setuptools import setup

ext_modules = [
    Pybind11Extension(
        "viterbicodec",
        ["viterbi.cpp", "viterbi_python.cpp"],
    ),
]

setup(ext_modules=ext_modules, cmdclass={"build_ext": build_ext})