From da37880bb7a80f5e8703641110f08187544caebf Mon Sep 17 00:00:00 2001
From: Jennifer Wei <jennifer.wei@omsf.io>
Date: Thu, 24 Apr 2025 22:25:20 +0000
Subject: [PATCH] Allow numpy>2 and support compute capability >9

---
 environment.yml | 2 +-
 setup.py        | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/environment.yml b/environment.yml
index 6a8a1c3..4489590 100644
--- a/environment.yml
+++ b/environment.yml
@@ -14,7 +14,7 @@ dependencies:
   - pdbfixer
   - pytorch-lightning
   - biopython
-  - numpy<2.0.0
+  - numpy
   - pandas
   - PyYAML
   - requests
diff --git a/setup.py b/setup.py
index 4873570..bd8c9cf 100644
--- a/setup.py
+++ b/setup.py
@@ -54,6 +54,7 @@ def get_cuda_bare_metal_version(cuda_dir):
 compute_capabilities = set([
     (5, 2), # Titan X
     (6, 1), # GeForce 1000-series
+    (9, 0), # Hopper
 ])
 
 compute_capabilities.add((7, 0))