From 780b6d0ee01ffbfac45f7ec5418bc08f2b166483 Mon Sep 17 00:00:00 2001
From: Antonio Sanchez <cantonios@google.com>
Date: Thu, 31 Aug 2023 15:43:36 -0700
Subject: [PATCH] Fix setup.py.

PiperOrigin-RevId: 561781166
---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index a0e9b205..031172e2 100644
--- 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",