try again

Author: ianardee

mindee/parsing/v2/dynamic_field.py

@@ -1,7 +1,7 @@
 from enum import Enum
+from importlib import import_module
 
 from mindee.error import MindeeApiV2Error
-from mindee.parsing import v2
 from mindee.parsing.common.string_dict import StringDict
 
 
@@ -34,13 +34,13 @@ def get_field_type(raw_response: StringDict, indent_level: int = 0) -> DynamicFi
     """Get appropriate field types."""
     if isinstance(raw_response, dict):
         if "value" in raw_response:
-            field_file = getattr(v2, "simple_field")
+            field_file = import_module("mindee.parsing.v2.simple_field")
             field_class = getattr(field_file, FieldType.SIMPLE.value)
         elif "items" in raw_response:
-            field_file = getattr(v2, "list_field")
+            field_file = import_module("mindee.parsing.v2.list_field")
             field_class = getattr(field_file, FieldType.LIST.value)
         elif "fields" in raw_response:
-            field_file = getattr(v2, "object_field")
+            field_file = import_module("mindee.parsing.v2.object_field")
             field_class = getattr(field_file, FieldType.OBJECT.value)
         else:
             raise MindeeApiV2Error(f"Unrecognized field format in {raw_response}.")

mindee/parsing/v2/inference_result_fields.py

@@ -1,7 +1,7 @@
 from typing import Dict
 
 from mindee.parsing.common.string_dict import StringDict
-from mindee.parsing.v2.dynamic_field import DynamicField, get_field_type
+from mindee.parsing.v2.dynamic_field import DynamicField, FieldType, get_field_type
 
 
 class InferenceResultFields(Dict[str, DynamicField]):
@@ -22,7 +22,7 @@ def __getattr__(self, item):
     def __str__(self) -> str:
         str_fields = ""
         for field_key, field_value in self.items():
-            if field_value.field_type == "SimpleField":
+            if field_value.field_type == FieldType.SIMPLE:
                 str_fields += f"\n:{field_key}: {field_value}"
             else:
                 str_fields += f"\n:{field_key}:{field_value}"