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",