From d0ce0532fcf18238eca68d307877b683fb134324 Mon Sep 17 00:00:00 2001 From: Alex Rogozhnikov Date: Tue, 30 Jun 2026 07:36:55 +0000 Subject: [PATCH 1/2] relax requirement on numpy --- requirements.in | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.in b/requirements.in index a51256d..105d673 100644 --- a/requirements.in +++ b/requirements.in @@ -20,7 +20,7 @@ modelcif>=1.0 # mmcif writing, confirmed to work currently latest 1.0 # computing, dl -numpy~=1.21 +numpy>=1.21 pandas[parquet,gcp,aws]~=2.1 pandera>=0.24 # pandera.pandas requires >= 0.24 numba>=0.59 From cbbd0a2586c6d40db1cc60f0ceb88b72bd7286fe Mon Sep 17 00:00:00 2001 From: Alex Rogozhnikov Date: Tue, 30 Jun 2026 07:45:38 +0000 Subject: [PATCH 2/2] allow_pickle = False (taming mypy) --- chai_lab/chai1.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/chai_lab/chai1.py b/chai_lab/chai1.py index 9170a17..ffd05af 100644 --- a/chai_lab/chai1.py +++ b/chai_lab/chai1.py @@ -1047,7 +1047,7 @@ def avg_per_token_1d(x): scores_out_path = output_dir.joinpath(f"scores.model_idx_{idx}.npz") - np.savez(scores_out_path, **get_scores(ranking_outputs)) + np.savez(scores_out_path, allow_pickle=False, **get_scores(ranking_outputs)) return StructureCandidates( cif_paths=cif_paths,