Avoid duplicate type bag

usethesource · Jan 10, 2024 · 337de0a · 337de0a
1 parent e3e1a47
commit 337de0a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...ain/java/io/usethesource/vallang/impl/persistent/PersistentHashIndexedBinaryRelation.java b/...ain/java/io/usethesource/vallang/impl/persistent/PersistentHashIndexedBinaryRelation.java
@@ -564,7 +564,7 @@ public ISet closure() {
       // this means no other types can be introduced other than the originals,
       // so iteration is no longer necessary to construct the new type bag
       keyTypeBag = AbstractTypeBag.of(keyType, result.size());
-      valTypeBag = AbstractTypeBag.of(valueType, result.size());
+      valTypeBag = keyTypeBag;
     }
     else {
       keyTypeBag = calcTypeBag(result, Map.Entry::getKey);
@@ -611,7 +611,7 @@ public ISet closureStar() {
       // this means no other types can be introduced other than the originals,
       // so iteration is no longer necessary to construct the new type bag
       keyTypeBag = AbstractTypeBag.of(keyType, result.size());
-      valTypeBag = AbstractTypeBag.of(valueType, result.size());
+      valTypeBag = keyTypeBag;
     }
     else {
       keyTypeBag = calcTypeBag(result, Map.Entry::getKey);