fix ml error msg

Author: ChristianBeilschmidt

geoengine/ml.py

@@ -103,10 +103,12 @@ def check_data_type(data_type: TypeProto, expected_type: RasterDataType, prefix:
         if not data_type.tensor_type:
             raise InputException('Only tensor input types are supported')
         elem_type = data_type.tensor_type.elem_type
-        if elem_type != RASTER_TYPE_TO_ONNX_TYPE[expected_type]:
+        expected_tensor_type = RASTER_TYPE_TO_ONNX_TYPE[expected_type]
+        if elem_type != expected_tensor_type:
             elem_type_str = tensor_dtype_to_string(elem_type)
+            expected_type_str = tensor_dtype_to_string(expected_tensor_type)
             raise InputException(f'Model {prefix} type `{elem_type_str}` does not match the '
-                                 f'expected type `{expected_type}`')
+                                 f'expected type `{expected_type_str}`')
 
     model_inputs = onnx_model.graph.input
     model_outputs = onnx_model.graph.output

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 `F64`'
+                'Model input type `TensorProto.FLOAT` does not match the expected type `TensorProto.DOUBLE`'
             )
 
             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 `I32`'
+                'Model output type `TensorProto.INT64` does not match the expected type `TensorProto.INT32`'
             )