From 197b6d46f3d8a0b052e58b29a0786edbe3a2dbd0 Mon Sep 17 00:00:00 2001
From: Kirill Zinovjev <kzinovjev@gmail.com>
Date: Tue, 22 Oct 2024 20:25:17 +0200
Subject: [PATCH] Provide "species" from model params to EMLEBase

---
 emle/models/_emle.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/emle/models/_emle.py b/emle/models/_emle.py
index 344faf1..7a23c0d 100644
--- a/emle/models/_emle.py
+++ b/emle/models/_emle.py
@@ -348,7 +348,7 @@ def __init__(
             aev_computer=self._aev_computer,
             aev_mask=aev_mask,
             alpha_mode=self._alpha_mode,
-            species=self._species,
+            species=params.get("species", self._species),
             device=device,
             dtype=dtype,
         )