1 files changed, 11 insertions, 1 deletions

diff --git a/subprojects/store-reasoning/src/main/java/tools/refinery/store/reasoning/translator/typehierarchy/TypeInfo.java b/subprojects/store-reasoning/src/main/java/tools/refinery/store/reasoning/translator/typehierarchy/TypeInfo.java
index 68ed4089..e6bdaff2 100644
--- a/subprojects/store-reasoning/src/main/java/tools/refinery/store/reasoning/translator/typehierarchy/TypeInfo.java
+++ b/subprojects/store-reasoning/src/main/java/tools/refinery/store/reasoning/translator/typehierarchy/TypeInfo.java
@@ -9,5 +9,15 @@ import tools.refinery.store.reasoning.representation.PartialRelation;
 
 import java.util.*;
 
-public record TypeInfo(Collection<PartialRelation> supertypes, boolean abstractType) {
+public record TypeInfo(Set<PartialRelation> supertypes, boolean abstractType) {
+        public TypeInfo(Collection<PartialRelation> supertypes, boolean abstractType) {
+                this(Set.copyOf(supertypes), abstractType);
+        }
+
+        public TypeInfo addSupertype(PartialRelation newSupertype) {
+                var newSupertypes = new ArrayList<PartialRelation>(supertypes.size() + 1);
+                newSupertypes.addAll(supertypes);
+                newSupertypes.add(newSupertype);
+                return new TypeInfo(newSupertypes, abstractType);
+        }
 }