diff --git a/mindee/v2/parsing/__init__.py b/mindee/v2/parsing/__init__.py index 8ca452fa..0d8702f5 100644 --- a/mindee/v2/parsing/__init__.py +++ b/mindee/v2/parsing/__init__.py @@ -2,6 +2,9 @@ from mindee.v2.parsing.error.error_response import ErrorResponse from mindee.v2.parsing.inference.base_inference import BaseInference from mindee.v2.parsing.inference.base_response import BaseResponse +from mindee.v2.parsing.inference.failed_inference_response import ( + FailedInferenceResponse, +) from mindee.v2.parsing.inference.inference_active_options import InferenceActiveOptions from mindee.v2.parsing.inference.inference_file import InferenceFile from mindee.v2.parsing.inference.inference_model import InferenceModel @@ -18,6 +21,7 @@ "ExtractionInference", "ExtractionResponse", "ExtractionResult", + "FailedInferenceResponse", "InferenceActiveOptions", "InferenceFile", "InferenceModel", diff --git a/mindee/v2/parsing/failed_inference_response.py b/mindee/v2/parsing/inference/failed_inference_response.py similarity index 94% rename from mindee/v2/parsing/failed_inference_response.py rename to mindee/v2/parsing/inference/failed_inference_response.py index ddca5645..8d481a7b 100644 --- a/mindee/v2/parsing/failed_inference_response.py +++ b/mindee/v2/parsing/inference/failed_inference_response.py @@ -2,7 +2,7 @@ from mindee.parsing.common.common_response import CommonResponse from mindee.parsing.common.string_dict import StringDict -from mindee.v2.parsing import ErrorResponse +from mindee.v2.parsing.error.error_response import ErrorResponse class FailedInferenceResponse(CommonResponse): diff --git a/mindee/v2/parsing/inference/field/simple_field.py b/mindee/v2/parsing/inference/field/simple_field.py index 65de5a51..f6e6c809 100644 --- a/mindee/v2/parsing/inference/field/simple_field.py +++ b/mindee/v2/parsing/inference/field/simple_field.py @@ -23,7 +23,7 @@ def string_value(self) -> str | None: return self.value @property - def number_value(self) -> float | None: + def float_value(self) -> float | None: """Retrieves a number field value as a float.""" if self.value is not None and not isinstance(self.value, float): raise ValueError("Value is not a number") diff --git a/tests/v2/parsing/test_failed_inference_response.py b/tests/v2/parsing/test_failed_inference_response.py index 611f721a..a6539f2a 100644 --- a/tests/v2/parsing/test_failed_inference_response.py +++ b/tests/v2/parsing/test_failed_inference_response.py @@ -3,7 +3,7 @@ import pytest -from mindee.v2.parsing.failed_inference_response import FailedInferenceResponse +from mindee.v2.parsing import FailedInferenceResponse from tests.utils import V2_DATA_DIR diff --git a/tests/v2/parsing/test_simple_field.py b/tests/v2/parsing/test_simple_field.py index 99fe3d80..4eb6e561 100644 --- a/tests/v2/parsing/test_simple_field.py +++ b/tests/v2/parsing/test_simple_field.py @@ -32,27 +32,27 @@ def test_raises_when_value_is_boolean(self): class TestSimpleFieldNumberValue: def test_returns_float_when_value_is_float(self): field = _make_field(3.14) - assert field.number_value == 3.14 + assert field.float_value == 3.14 def test_returns_float_when_value_is_int(self): # Integers are coerced to float in the constructor field = SimpleField({"value": 42}) - assert field.number_value == 42.0 - assert isinstance(field.number_value, float) + assert field.float_value == 42.0 + assert isinstance(field.float_value, float) def test_returns_none_when_value_is_none(self): field = _make_field(None) - assert field.number_value is None + assert field.float_value is None def test_raises_when_value_is_string(self): field = _make_field("42") with pytest.raises(ValueError, match="Value is not a number"): - _ = field.number_value + _ = field.float_value def test_raises_when_value_is_boolean(self): field = _make_field(True) with pytest.raises(ValueError, match="Value is not a number"): - _ = field.number_value + _ = field.float_value @pytest.mark.v2