fix test

Author: ChristianBeilschmidt

setup.cfg

@@ -34,7 +34,7 @@ install_requires =
     websockets >= 10.0,<11
     xarray >=0.19,<2024.12
     urllib3 >= 2.0, < 2.3
-    pydantic >= 2, < 2.11
+    pydantic >= 2.10.6, < 2.11
     skl2onnx >=1.17,<2
 
 [options.extras_require]

tests/test_ml.py

@@ -106,7 +106,7 @@ def test_uploading_onnx_model(self):
                 )
             self.assertEqual(
                 str(exception.exception),
-                'Model input type `TensorProto.FLOAT` does not match the expected type `RasterDataType.F64`'
+                'Model input type `TensorProto.FLOAT` does not match the expected type `F64`'
             )
 
             with self.assertRaises(ge.InputException) as exception:
@@ -126,5 +126,5 @@ def test_uploading_onnx_model(self):
                 )
             self.assertEqual(
                 str(exception.exception),
-                'Model output type `TensorProto.INT64` does not match the expected type `RasterDataType.I32`'
+                'Model output type `TensorProto.INT64` does not match the expected type `I32`'
             )