fix mindspore.ops.norm kw param

Author: rogerwwww

pygmtools/mindspore_backend.py

@@ -31,12 +31,12 @@ def _ms_max(*args, keep_dims=False, **kwargs):
                      'Mindspore version to GitHub issues.')
 
 _norm_signature = inspect.signature(mindspore.ops.norm)
-if 'axis' in _norm_signature.parameters and 'p' in _norm_signature.parameters:
-    def _ms_norm(*args, p=None, axis=None, **kwargs):
-        return mindspore.ops.norm(*args, p=p, axis=axis, **kwargs)
-elif 'dim' in _norm_signature.parameters and 'ord' in _norm_signature.parameters:
-    def _ms_norm(*args, p=None, axis=None, **kwargs):
-        return mindspore.ops.norm(*args, ord=p, dim=axis, **kwargs)
+if 'axis' in _norm_signature.parameters and 'p' in _norm_signature.parameters and 'keep_dims' in _norm_signature.parameters:
+    def _ms_norm(*args, p=None, axis=None, keep_dims=False, **kwargs):
+        return mindspore.ops.norm(*args, p=p, axis=axis, keep_dims=keep_dims, **kwargs)
+elif 'dim' in _norm_signature.parameters and 'ord' in _norm_signature.parameters and 'keepdim' in _norm_signature.parameters:
+    def _ms_norm(*args, p=None, axis=None, keepdim=False, **kwargs):
+        return mindspore.ops.norm(*args, ord=p, dim=axis, keepdim=keep_dims, **kwargs)
 else:
     raise ValueError('Mindspore function mindspore.ops.norm has unsupported signature. It is likely you are '
                      'working with a new Mindspore version which breaks backward compatibility. Please report your '