[BugFix] Fix tensorizer extra in setup.py (#4072)

This commit is contained in:
Nick Hill 2024-04-14 22:12:42 +01:00 committed by GitHub
parent 2cd6b4f362
commit 563c54f760
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -406,7 +406,7 @@ setup(
install_requires=get_requirements(), install_requires=get_requirements(),
ext_modules=ext_modules, ext_modules=ext_modules,
extras_require={ extras_require={
"optional": ["tensorizer==2.9.0a1"], "tensorizer": ["tensorizer==2.9.0a1"],
}, },
cmdclass={"build_ext": cmake_build_ext} if not _is_neuron() else {}, cmdclass={"build_ext": cmake_build_ext} if not _is_neuron() else {},
package_data=package_data, package_data=package_data,