Fixed __pycache__ and __init__ imports for full_featurizer

lrcfmd · Oct 25, 2024 · ee247d7 · ee247d7
1 parent ab6dc38
commit ee247d7
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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()])