From 0d641c86a0c12811889dd6c12937324236915f44 Mon Sep 17 00:00:00 2001
From: Brian Bargh <brian.bargh@gmail.com>
Date: Wed, 8 May 2024 21:35:50 +0000
Subject: [PATCH] Enable allow_pickle to be able to specify splits.

---
 torchmdnet/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torchmdnet/utils.py b/torchmdnet/utils.py
index 7ed94301..d01c3fb6 100644
--- a/torchmdnet/utils.py
+++ b/torchmdnet/utils.py
@@ -247,7 +247,7 @@ def make_splits(
     order=None,
 ):
     if splits is not None:
-        splits = np.load(splits)
+        splits = np.load(splits, allow_pickle=True)
         idx_train = splits["idx_train"]
         idx_val = splits["idx_val"]
         idx_test = splits["idx_test"]