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

This commit is contained in:
Nick Hill authored and GitHub committed 2024-04-14 14:12:42 -07:00
1 parent 2cd6b4f362
commit 563c54f760
1 file changed
+1 -1
+1 -1
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,