From e58dff56ba2ba4a9bee5500497bfe202dc5eca11 Mon Sep 17 00:00:00 2001
From: Matthew Evans <7916000+ml-evs@users.noreply.github.com>
Date: Mon, 22 Apr 2024 18:40:14 +0200
Subject: [PATCH] Suppress pymatgen element data warnings when imputing (#934)

---
 matminer/utils/data.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/matminer/utils/data.py b/matminer/utils/data.py
index de8aea73..19306667 100644
--- a/matminer/utils/data.py
+++ b/matminer/utils/data.py
@@ -357,7 +357,12 @@ def get_elemental_property(self, elem, property_name):
             block_key = {"s": 1.0, "p": 2.0, "d": 3.0, "f": 3.0}
             return block_key[getattr(elem, property_name)]
         else:
-            value = getattr(elem, property_name)
+            # Suppress pymatgen warnings for missing properties if we are going to impute anyway
+            with warnings.catch_warnings():
+                if self.impute_nan:
+                    warnings.simplefilter("ignore", category=UserWarning)
+                value = getattr(elem, property_name)
+
             if self.impute_nan:
                 if value and not pd.isnull(value):
                     return value