Merge pull request #151 from IDEA-Research/fix/preview_grounding

fix(app): preview grounding annotations with filter
IDEA-Research · Apr 10, 2024 · 1f4c99a · 1f4c99a
2 parents 90a2c81 + d14391d
commit 1f4c99a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/components/src/Annotator/preview.tsx b/packages/components/src/Annotator/preview.tsx
@@ -358,7 +358,7 @@ const Preview: React.FC<PreviewProps> = (props) => {
   }
 
   const getHighlightWords = () => {
-    const objects = list[current].objects;
+    const objects = objectsFilter ? objectsFilter(list[current]) : list[current].objects;
     return categories
       .filter((category) => objects.find((obj: any) => obj.categoryId === category.id))
       .map((category) => {