[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(),
ext_modules=ext_modules,
extras_require={
"optional": ["tensorizer==2.9.0a1"],
"tensorizer": ["tensorizer==2.9.0a1"],
},
cmdclass={"build_ext": cmake_build_ext} if not _is_neuron() else {},
package_data=package_data,