From 0775d194fce5c70d3a470f85c486475fdb7026d4 Mon Sep 17 00:00:00 2001
From: Nick Hill <nick@nickhill.org>
Date: Sun, 14 Apr 2024 12:20:55 -0700
Subject: [PATCH] [BugFix] Fix tensorizer extra in setup.py

extras_require is meant to be a dict of the names of some extra features to lists of dependencies that each requires.
---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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,