Fixed unit tests

Author: Tim-Kerr

libs/labelbox/src/labelbox/schema/tool_building/classification.py

@@ -95,8 +95,6 @@ def __post_init__(self):
 
     @classmethod
     def from_dict(cls, dictionary: Dict[str, Any]) -> "Classification":
-        print('attributes:')
-        print(dictionary.get("attributes", None))
         return cls(
             class_type=Classification.Type(dictionary["type"]),
             name=dictionary["name"],
@@ -109,7 +107,7 @@ def from_dict(cls, dictionary: Dict[str, Any]) -> "Classification":
             schema_id=dictionary.get("schemaNodeId", None),
             feature_schema_id=dictionary.get("featureSchemaId", None),
             scope=cls.Scope(dictionary.get("scope", cls.Scope.GLOBAL)),
-            attributes=FeatureSchemaAttributes(dictionary.get("attributes", None)),
+            attributes=dictionary.get("attributes", None),
         )
 
     def asdict(self, is_subclass: bool = False) -> Dict[str, Any]:

libs/labelbox/tests/integration/test_ontology.py

@@ -225,7 +225,6 @@ def feature_schema_cat_normalized(name_for_read):
         "name": name_for_read,
         "color": "black",
         "classifications": [],
-        "attributes": [] # TODO tkerr: Finish
     }
 
 

libs/labelbox/tests/unit/test_unit_ontology.py

@@ -15,6 +15,7 @@
             "color": "#FF0000",
             "tool": "polygon",
             "classifications": [],
+            "attributes": None
         },
         {
             "schemaNodeId": None,
@@ -24,6 +25,7 @@
             "color": "#FF0000",
             "tool": "superpixel",
             "classifications": [],
+            "attributes": None
         },
         {
             "schemaNodeId": None,
@@ -32,6 +34,7 @@
             "name": "bbox",
             "color": "#FF0000",
             "tool": "rectangle",
+            "attributes": None,
             "classifications": [
                 {
                     "schemaNodeId": None,
@@ -56,6 +59,7 @@
                                     "name": "nested nested text",
                                     "type": "text",
                                     "options": [],
+                                    "attributes": None
                                 }
                             ],
                         },
@@ -67,6 +71,7 @@
                             "options": [],
                         },
                     ],
+                    "attributes": None
                 },
                 {
                     "schemaNodeId": None,
@@ -76,6 +81,7 @@
                     "name": "nested text",
                     "type": "text",
                     "options": [],
+                    "attributes": None
                 },
             ],
         },
@@ -87,6 +93,7 @@
             "color": "#FF0000",
             "tool": "point",
             "classifications": [],
+            "attributes": None
         },
         {
             "schemaNodeId": None,
@@ -96,6 +103,7 @@
             "color": "#FF0000",
             "tool": "line",
             "classifications": [],
+            "attributes": None
         },
         {
             "schemaNodeId": None,
@@ -105,6 +113,7 @@
             "color": "#FF0000",
             "tool": "named-entity",
             "classifications": [],
+            "attributes": None
         },
     ],
     "classifications": [
@@ -117,6 +126,7 @@
             "type": "radio",
             "scope": "global",
             "uiMode": "searchable",
+            "attributes": None,
             "options": [
                 {
                     "schemaNodeId": None,

libs/labelbox/tests/unit/test_unit_prompt_issue_tool.py

@@ -47,6 +47,7 @@ def test_as_dict():
                 "schemaNodeId": None,
                 "featureSchemaId": None,
                 "scope": "global",
+                "attributes": None
             }
         ],
         "color": None,