add numpy 2 hotfix to main

main.py

@@ -261,6 +261,38 @@
 ]
 
 
+def _update_numpy_base_dependencies(fn, record, instructions):
+    depends = record.get("depends", [])
+    updated = False
+    for i, dep in enumerate(depends):
+        if dep.split()[0] == "numpy-base" and '==' not in dep and "1." not in dep and '<' not in dep:
+            depends[i] = dep + ",<2.0a0"
+            updated = True
+        else:
+            depends[i] = "numpy-base <2.0a0"
+            updated = True
+    if updated:
+        instructions["packages"][fn]['depends'] = depends
+    return updated
+
+
+def _update_numpy_dependencies(fn, record, instructions):
+    depends = record.get("depends", [])
+    updated = False
+    for i, dep in enumerate(depends):
+        if dep.split()[0] == "numpy":
+            # If there's no upper bound and the dependency isn't pinned to a specific version
+            if '==' not in dep and "1." not in dep and '<' not in dep:
+                depends[i] = dep + ",<2.0a0"
+                updated = True
+            elif dep == "numpy":
+                depends[i] = "numpy <2.0a0"
+                updated = True
+    if updated:
+        instructions["packages"][fn]['depends'] = depends
+    return updated
+
+
 def _replace_vc_features_with_vc_pkg_deps(name, record, depends):
     python_vc_deps = {
         "2.6": "vc 9.*",
@@ -671,6 +703,18 @@ def patch_record_in_place(fn, record, subdir):
                 depends[i] = depends[i].replace(">=1.21.5,", ">=1.21.2,")
                 break
 
+    # to update dependencies for preperation for numpy 2.0.0
+    numpy_instructions = {
+        "patch_instructions_version": 1,
+        "packages": defaultdict(dict),
+        "revoke": [],
+        "remove": [],
+    }
+    if name == "numpy":
+        _update_numpy_dependencies(fn, record, numpy_instructions)
+    elif name == "numpy-base":
+        _update_numpy_base_dependencies(fn, record, numpy_instructions)
+
     ###########
     # pytorch #
     ###########