From 4b002e3de8a50e83155e1ec80fbf0d2707108763 Mon Sep 17 00:00:00 2001
From: Lucas Attia <43359170+lucasattia@users.noreply.github.com>
Date: Thu, 5 Dec 2024 14:17:23 -0500
Subject: [PATCH] Update utils.py

updating SUNSET_SPECIES_TABLE to reflect the new naming convention
---
 src/NanoParticleTools/machine_learning/data/utils.py | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/src/NanoParticleTools/machine_learning/data/utils.py b/src/NanoParticleTools/machine_learning/data/utils.py
index d3c67c3..57a5caf 100644
--- 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"])
 }