Update utils.py

updating SUNSET_SPECIES_TABLE to reflect the new naming convention
BlauGroup · Dec 5, 2024 · 4b002e3 · 4b002e3
1 parent a32e22f
commit 4b002e3
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/src/NanoParticleTools/machine_learning/data/utils.py b/src/NanoParticleTools/machine_learning/data/utils.py
@@ -5,12 +5,10 @@
 import os
 
 SUNSET_SPECIES_TABLE = {
-    1: sorted(["Yb", "Er", "Xsurfacesix"]),
-    2: sorted(["Yb", "Er"]),
-    3: sorted(["Yb", "Er", "Xsurfacesix", "Tm"]),
-    4: sorted(["Yb", "Er"]),
-    5: sorted(["Yb", "Er", "Nd"]),
-    6: sorted(['Yb', 'Er', "Xsurfacesix", 'Tm', 'Nd', 'Ho', 'Eu', 'Sm', 'Dy'])
+    1: sorted(["Yb", "Er", "Nd"]),
+    2: sorted(["Yb", "Er", "Xsurfacesix"]),
+    3: sorted(["Yb", "Er"]),
+    4: sorted(["Yb", "Er", "Xsurfacesix", "Tm"])
 }