Upgrade ml-dtypes in environment.yml to fix conflict

Author: vkarampudi

environment.yml

@@ -5,6 +5,7 @@ dependencies:
   - python=3.11
   - pip
   - pip:
+    - --index-url https://pypi.org/simple
     - absl-py==2.4.0
     - altgraph==0.17.5
     - astunparse==1.6.3
@@ -25,7 +26,7 @@ dependencies:
     - markdown-it-py==4.0.0
     - markupsafe==3.0.3
     - mdurl==0.1.2
-    - ml-dtypes==0.4.1
+    - ml-dtypes==0.5.1
     - namex==0.1.0
     - numpy==1.26.4
     - opt-einsum==3.4.0