From f6c0c16932e59373b631ceca686d810832a66b5d Mon Sep 17 00:00:00 2001
From: RaulPPealez <raulppelaez@gmail.com>
Date: Wed, 28 Aug 2024 16:56:04 +0200
Subject: [PATCH] Remove dependency on ase in MACEOFF

---
 torchmdnet/datasets/maceoff.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/torchmdnet/datasets/maceoff.py b/torchmdnet/datasets/maceoff.py
index cb0d7f75..5691c596 100644
--- a/torchmdnet/datasets/maceoff.py
+++ b/torchmdnet/datasets/maceoff.py
@@ -3,7 +3,7 @@
 # (See accompanying file README.md file or copy at http://opensource.org/licenses/MIT)
 
 import hashlib
-from ase.data import atomic_numbers
+from torchmdnet.utils import ATOMIC_NUMBERS
 import numpy as np
 import os
 import torch as pt
@@ -44,7 +44,7 @@ def parse_maceoff_tar(tar_file):
                     counter = 2
                     continue
                 el, x, y, z, fx, fy, fz, _, _, _ = line.split()
-                numbers.append(atomic_numbers[el])
+                numbers.append(ATOMIC_NUMBERS[el])
                 positions.append([float(x), float(y), float(z)])
                 forces.append([float(fx), float(fy), float(fz)])
                 counter += 1