Fix linting errors.

kaueltzen · Oct 7, 2024 · 198bec5 · 198bec5
1 parent 5106b84
commit 198bec5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/matminer/featurizers/site/tests/test_fingerprint.py b/matminer/featurizers/site/tests/test_fingerprint.py
@@ -277,7 +277,7 @@ def test_crystal_nn_fingerprint(self):
 
     def test_chemenv_site_fingerprint(self):
         cefp = ChemEnvSiteFingerprint.from_preset("multi_weights")
-        implemented_cetypes = set([gg.ce_symbol for gg in cefp.lgf.allcg.get_implemented_geometries()])
+        implemented_cetypes = {gg.ce_symbol for gg in cefp.lgf.allcg.get_implemented_geometries()}
         assert set(cefp.cetypes).difference(implemented_cetypes) == set()  # Added after issue #945
         l = cefp.feature_labels()
         cevals = cefp.featurize(self.sc, 0)