Overload '+' operator to concatenate ConfigSet objects

libAtoms · Oct 3, 2024 · 9bb5dc2 · 9bb5dc2
1 parent 6655d94
commit 9bb5dc2
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/wfl/configset.py b/wfl/configset.py
@@ -388,6 +388,12 @@ def _flat_iter(items):
                     yield at
 
 
+    def __add__(self, other):
+        if not isinstance(other, ConfigSet):
+            raise TypeError(f"unsupported operand type(s) for +: 'ConfigSet' and '{type(other)}'")
+        return ConfigSet([self, other])
+
+
 class OutputSpec:
     """Abstraction for writing to a ConfigSet, preserving tree structure.