diff --git a/setup.py b/setup.py index 813321efe796..19a9150ad2e6 100644 --- a/setup.py +++ b/setup.py @@ -406,7 +406,7 @@ def _read_requirements(filename: str) -> List[str]: 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,