shim/setup.py

"""Tiny package whose only purpose is to drop transformers_shim.pth into
site-packages. Python loads .pth files at interpreter startup, before the
HF inference toolkit's `webservice_starlette` imports — so we get a hook
to patch transformers BEFORE the toolkit's old API call tries to import
names that 5.x dropped.

setuptools' data_files paths are relative to the install prefix ("data"
scheme path). To land the .pth in purelib (site-packages) we compute the
relative path from data → purelib at build time. Pip builds the wheel
on the target machine when given `./shim` so this resolves correctly.
"""
import os
from sysconfig import get_paths
from setuptools import setup

paths = get_paths()
pth_dest = os.path.relpath(paths["purelib"], paths["data"])

setup(
    name="transformers-shim",
    version="0.1.0",
    py_modules=["transformers_shim_runtime"],
    data_files=[(pth_dest, ["transformers_shim.pth"])],
)