Added from_dict method to the FeatureSchemaAttribute data class

Author: Tim-Kerr

libs/labelbox/src/labelbox/schema/ontology.py

@@ -123,10 +123,7 @@ def from_dict(cls, dictionary: Dict[str, Any]) -> Dict[str, Any]:
             ],
             color=dictionary["color"],
             attributes=[
-                FeatureSchemaAttribute(
-                    attributeName=attr["attributeName"],
-                    attributeValue=attr["attributeValue"],
-                )
+                FeatureSchemaAttribute.from_dict(attr)
                 for attr in dictionary.get("attributes", []) or []
             ]
             if dictionary.get("attributes")

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

@@ -114,10 +114,7 @@ def from_dict(cls, dictionary: Dict[str, Any]) -> "Classification":
             feature_schema_id=dictionary.get("featureSchemaId", None),
             scope=cls.Scope(dictionary.get("scope", cls.Scope.GLOBAL)),
             attributes=[
-                FeatureSchemaAttribute(
-                    attributeName=attr["attributeName"],
-                    attributeValue=attr["attributeValue"],
-                )
+                FeatureSchemaAttribute.from_dict(attr)
                 for attr in dictionary.get("attributes", []) or []
             ]
             if dictionary.get("attributes")

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

@@ -5,6 +5,8 @@
 
 from dataclasses import dataclass
 
+from typing import Any, Dict, List
+
 
 @dataclass
 class FeatureSchemaAttribute:
@@ -17,6 +19,13 @@ def asdict(self):
             "attributeValue": self.attributeValue,
         }
 
+    @classmethod
+    def from_dict(cls, dictionary: Dict[str, Any]) -> "FeatureSchemaAttribute":
+        return cls(
+            attributeName=dictionary["attributeName"],
+            attributeValue=dictionary["attributeValue"],
+        )
+
 
 FeatureSchemaAttribute = Annotated[FeatureSchemaAttribute, Field()]