Always install the correct version with pip_build

Mirror keras-team/keras-core#595 for keras-nlp
mattdangerw · Jul 24, 2023 · 36b1b65 · 36b1b65
1 parent 85d7c53
commit 36b1b65
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pip_build.py b/pip_build.py
@@ -88,7 +88,7 @@ def build():
 
         # Find the .whl file path
         for fname in os.listdir(dist_directory):
-            if fname.endswith(".whl"):
+            if __version__ in fname and fname.endswith(".whl"):
                 whl_path = os.path.abspath(os.path.join(dist_directory, fname))
         print(f"Build successful. Wheel file available at {whl_path}")
     finally: