add slots (#51)

allenai · Oct 11, 2023 · 309f7c9 · 309f7c9
1 parent 3a318fc
commit 309f7c9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/papermage/magelib/entity.py b/papermage/magelib/entity.py
@@ -16,6 +16,8 @@
 
 
 class Entity:
+    __slots__ = ["spans", "boxes", "images", "metadata", "_id", "_doc"]
+
     def __init__(
         self,
         spans: Optional[List[Span]] = None,

diff --git a/papermage/magelib/image.py b/papermage/magelib/image.py
@@ -17,6 +17,8 @@
 
 
 class Image:
+    __slots__ = ["_pilimage"]
+
     def __init__(self):
         self._pilimage = None