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

563c54f7 · Nick Hill · GitHub · 2cd6b4f3 · 563c54f7
Unverified Commit 563c54f7 authored Apr 14, 2024 by Nick Hill Committed by GitHub Apr 14, 2024
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

setup.py setup.py +1 -1

No files found.
--- a/setup.py
+++ b/setup.py
@@ -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,