Unverified Commit 563c54f7 authored by Nick Hill's avatar Nick Hill Committed by GitHub
Browse files

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

parent 2cd6b4f3
...@@ -406,7 +406,7 @@ setup( ...@@ -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,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment