fix: FragmentClassifier.byPredefineType returns a set of strings (#266)

ThatOpen · Jan 16, 2024 · b836e28 · b836e28
1 parent 91d69a9
commit b836e28
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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));
         }
       }
     }