From ee247d797fa31339c3b37e1fbd9e3acba1d56794 Mon Sep 17 00:00:00 2001 From: Cameron Hargreaves Date: Sat, 26 Oct 2024 00:28:40 +0200 Subject: [PATCH] Fixed __pycache__ and __init__ imports for full_featurizer --- ElMD/ElMD.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/ElMD/ElMD.py b/ElMD/ElMD.py index 6ce6867..e0a0e39 100644 --- a/ElMD/ElMD.py +++ b/ElMD/ElMD.py @@ -220,6 +220,10 @@ def full_feature_vector(self, positional_encode=False, min_freq=1e-4): full_features = [] for f in feature_dicts: + # Fixes __pycache__ and __init__ cases + if '__' in f: + continue + d, _ = _get_periodic_tab(f) vectors = np.array([d[el] for el in self.normed_composition.keys()])