is_created_by bug fix

asyml · Aug 26, 2022 · ab97ae9 · ab97ae9
1 parent 34d57a0
commit ab97ae9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/forte/data/base_pack.py b/forte/data/base_pack.py
@@ -806,8 +806,10 @@ def is_created_by(
         if isinstance(components, str):
             components = [components]
 
+        entry_tid = entry.tid if isinstance(entry, Entry) else entry
+
         for c in components:
-            if entry in self._creation_records[c]:
+            if entry_tid in self._creation_records[c]:
                 break
         else:
             # The entry not created by any of these components.