fixed union method

Author: alexhroom

RATapi/classlist.py

@@ -295,10 +295,14 @@ def union(self, other: Sequence[T]) -> None:
         """Extend the ClassList by a sequence, ignoring input items with names that already exist."""
         if other and not (isinstance(other, Sequence) and not isinstance(other, str)):
             other = [other]
-        if hasattr(other, self.name_field):
-            other = (item for item in other if getattr(other, self.name_field) not in self.get_names())
 
-        self.extend(other)
+        self.extend(
+            [
+                item
+                for item in other
+                if hasattr(item, self.name_field) and getattr(item, self.name_field) not in self.get_names()
+            ]
+        )
 
     def set_fields(self, index: int, **kwargs) -> None:
         """Assign the values of an existing object's attributes using keyword arguments."""

RATapi/utils/orso.py

@@ -7,7 +7,7 @@
 from orsopy.fileio import load_orso
 
 from RATapi import ClassList, Project
-from RATapi.models import AbsorptionLayer, Background, Contrast, Data, Layer, Parameter, Resolution
+from RATapi.models import AbsorptionLayer, Contrast, Data, Layer, Parameter, Resolution
 
 
 def load_ort_data(filepath: str) -> Union[Data, list[Data]]:
@@ -167,12 +167,13 @@ def orso_model_to_rat(model: orsopy.fileio.model_language.SampleModel | str, abs
         parameters.union(layer_params)
         layers.union(layer)
 
-
-    return ORSOSample(bulk_in=bulk_in,
-                      bulk_out=bulk_out,
-                      parameters=parameters,
-                      layers=layers,
-                      model=[layer.material.formula for layer in stack[1:-1]])
+    return ORSOSample(
+        bulk_in=bulk_in,
+        bulk_out=bulk_out,
+        parameters=parameters,
+        layers=layers,
+        model=[layer.material.formula for layer in stack[1:-1]],
+    )
 
 
 def orso_layer_to_rat_layer(