From b836e28016ce3a10b8b2f0ce89654a003aec8a83 Mon Sep 17 00:00:00 2001
From: Juan Hoyos <84806689+HoyosJuan@users.noreply.github.com>
Date: Tue, 16 Jan 2024 14:39:02 -0500
Subject: [PATCH] fix: FragmentClassifier.byPredefineType returns a set of
 strings (#266)

---
 resources/openbim-components.js           | 2 +-
 src/fragments/FragmentClassifier/index.ts | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/resources/openbim-components.js b/resources/openbim-components.js
index 49e0e2342..521f86561 100644
--- a/resources/openbim-components.js
+++ b/resources/openbim-components.js
@@ -104403,7 +104403,7 @@ class FragmentClassifier extends Component {
                         currentType[fragmentID] = new Set();
                     }
                     const currentFragment = currentType[fragmentID];
-                    currentFragment.add(entity.expressID);
+                    currentFragment.add(String(entity.expressID));
                 }
             }
         }
diff --git a/src/fragments/FragmentClassifier/index.ts b/src/fragments/FragmentClassifier/index.ts
index 5999a62bc..f22e312c2 100644
--- a/src/fragments/FragmentClassifier/index.ts
+++ b/src/fragments/FragmentClassifier/index.ts
@@ -9,7 +9,7 @@ import { FragmentManager } from "../FragmentManager";
 
 export interface Classification {
   [system: string]: {
-    [className: string]: { [fragmentID: string]: Set<string> };
+    [className: string]: FragmentIdMap;
   };
 }
 
@@ -201,7 +201,7 @@ export class FragmentClassifier
             currentType[fragmentID] = new Set<string>();
           }
           const currentFragment = currentType[fragmentID];
-          currentFragment.add(entity.expressID);
+          currentFragment.add(String(entity.expressID));
         }
       }
     }