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 #
  ###########