Add py.typed for mypy use

Allows mypy to type check when used in other packages via PEP 561.
This commit is contained in:
Hayk Martiros 2023-03-26 23:37:52 +00:00
parent 754c9f8f64
commit 8d44cbbc64
2 changed files with 4 additions and 0 deletions

1
riffusion/py.typed Normal file
View File

@ -0,0 +1 @@
# https://peps.python.org/pep-0561/

View File

@ -28,6 +28,9 @@ setup(
"transformers", "transformers",
], ],
install_requires=packages, install_requires=packages,
package_data={
"riffusion": ["py.typed"],
},
classifiers=[ classifiers=[
"Development Status :: 4 - Beta", "Development Status :: 4 - Beta",
"Intended Audience :: Developers", "Intended Audience :: Developers",