Fix setup.py.

PiperOrigin-RevId: 561781166
jax-ml · Aug 31, 2023 · 780b6d0 · 780b6d0
1 parent 28cc246
commit 780b6d0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -52,7 +52,7 @@ def find_package_modules(self, package, package_dir):
 setup(
     ext_modules=[
         Pybind11Extension(
-            "ml_dtypes._ml_dtypes_lib",
+            "ml_dtypes._ml_dtypes_ext",
             [
                 "ml_dtypes/_src/dtypes.cc",
                 "ml_dtypes/_src/numpy.cc",