diff --git a/setup.py b/setup.py index 03999842..4ceba2f6 100644 --- a/setup.py +++ b/setup.py @@ -15,6 +15,9 @@ setup(name='tinygrad', long_description=long_description, long_description_content_type='text/markdown', packages = ['tinygrad', 'tinygrad.codegen', 'tinygrad.nn', 'tinygrad.renderer', 'tinygrad.runtime', 'tinygrad.shape'], + package_data = { + 'tinygrad': ['py.typed'], + }, classifiers=[ "Programming Language :: Python :: 3", "License :: OSI Approved :: MIT License" diff --git a/tinygrad/py.typed b/tinygrad/py.typed new file mode 100644 index 00000000..e69de29b