diff --git a/coremltools/py.typed b/coremltools/py.typed new file mode 100644 index 000000000..8b1378917 --- /dev/null +++ b/coremltools/py.typed @@ -0,0 +1 @@ + diff --git a/setup.py b/setup.py index d61076b7d..48f0f7bda 100755 --- a/setup.py +++ b/setup.py @@ -66,7 +66,8 @@ url="https://github.com/apple/coremltools", packages=find_packages(), package_data={ - "": ["LICENSE.txt", "README.md", "libcaffeconverter.so", "libcoremlpython.so"] + "": ["LICENSE.txt", "README.md", "libcaffeconverter.so", "libcoremlpython.so"], + "coremltools": ["py.typed"] }, install_requires=[ "numpy >= 1.14.5, < 1.20",