Fix marshmallow 4.x compatibility: Replace FieldInstanceResolutionError with ValueError

Co-authored-by: kshitij-microsoft <[email protected]>

Author: Copilot

sdk/ml/azure-ai-ml/azure/ai/ml/_schema/core/fields.py

@@ -17,7 +17,7 @@
 from marshmallow import RAISE, fields
 from marshmallow.exceptions import ValidationError
 from marshmallow.fields import Field, Nested
-from marshmallow.utils import FieldInstanceResolutionError, from_iso_datetime, resolve_field_instance
+from marshmallow.utils import from_iso_datetime, resolve_field_instance
 
 from ..._utils._arm_id_utils import AMLVersionedArmId, is_ARM_id_for_resource, parse_name_label, parse_name_version
 from ..._utils._experimental import _is_warning_cached
@@ -439,7 +439,7 @@ def __init__(self, union_fields: List[fields.Field], is_strict=False, **kwargs):
             self._union_fields = [resolve_field_instance(cls_or_instance) for cls_or_instance in union_fields]
             # TODO: make serialization/de-serialization work in the same way as json schema when is_strict is True
             self.is_strict = is_strict  # S\When True, combine fields with oneOf instead of anyOf at schema generation
-        except FieldInstanceResolutionError as error:
+        except ValueError as error:
             raise ValueError(
                 'Elements of "union_fields" must be subclasses or instances of marshmallow.base.FieldABC.'
             ) from error
@@ -876,7 +876,7 @@ def __init__(self, experimental_field: fields.Field, **kwargs):
         try:
             self._experimental_field = resolve_field_instance(experimental_field)
             self.required = experimental_field.required
-        except FieldInstanceResolutionError as error:
+        except ValueError as error:
             raise ValueError(
                 '"experimental_field" must be subclasses or instances of marshmallow.base.FieldABC.'
             ) from error

sdk/ml/azure-ai-ml/test_marshmallow_migration.py

@@ -57,7 +57,7 @@ def test_basic_imports():
         from marshmallow.schema import SchemaMeta
         from marshmallow.exceptions import ValidationError as SchemaValidationError
         from marshmallow.fields import Field, Nested
-        from marshmallow.utils import FieldInstanceResolutionError, from_iso_datetime, resolve_field_instance
+        from marshmallow.utils import from_iso_datetime, resolve_field_instance
 
         print("✓ All basic imports successful")
         return True