Box default attributes

asyml · Jun 30, 2022 · 8b8d74d · 8b8d74d
1 parent a9998d6
commit 8b8d74d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/forte/data/ontology/top.py b/forte/data/ontology/top.py
@@ -1096,8 +1096,8 @@ class Box(Region):
     def __init__(
         self,
         pack: PackType,
-        cy: int,
-        cx: int,
+        cy: int = 0,
+        cx: int = 0,
         height: int = 1,
         width: int = 1,
         image_payload_idx: int = 0,

diff --git a/tests/forte/data/data_store_test.py b/tests/forte/data/data_store_test.py
@@ -1101,7 +1101,7 @@ def test_get(self):
                 "forte.data.ontology.top.Annotation", range_annotation=(1, 20)
             )
         )
-        self.assertEqual(len(instances), 4)
+        self.assertEqual(len(instances), 2)
 
         # get groups with subclasses
         instances = list(self.data_store.get("forte.data.ontology.top.Group"))