diff --git a/setup.py b/setup.py index c17b925e9a..005926b155 100644 --- a/setup.py +++ b/setup.py @@ -16,6 +16,7 @@ setup(name='tinygrad', long_description_content_type='text/markdown', packages = ['tinygrad', 'tinygrad.runtime.autogen', 'tinygrad.codegen', 'tinygrad.nn', 'tinygrad.renderer', 'tinygrad.engine', 'tinygrad.runtime', 'tinygrad.runtime.support', 'tinygrad.runtime.graph', '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 0000000000..e69de29bb2