Add py.typed for mypy use

Allows mypy to type check when used in other packages via PEP 561.
2023-03-26 23:37:52 +00:00 · 2023-03-26 23:37:52 +00:00 · 8d44cbbc64
parent 754c9f8f64
commit 8d44cbbc64
2 changed files with 4 additions and 0 deletions
--- a/riffusion/py.typed
+++ b/riffusion/py.typed
@ -0,0 +1 @@
+# https://peps.python.org/pep-0561/
--- a/setup.py
+++ b/setup.py
@ -28,6 +28,9 @@ setup(
        "transformers",
    ],
    install_requires=packages,
+    package_data={
+        "riffusion": ["py.typed"],
+    },
    classifiers=[
        "Development Status :: 4 - Beta",
        "Intended Audience :: Developers",