diff --git a/learnware/specification/regular/image/rkme.py b/learnware/specification/regular/image/rkme.py index 75086c2..5bf57fd 100644 --- a/learnware/specification/regular/image/rkme.py +++ b/learnware/specification/regular/image/rkme.py @@ -98,9 +98,7 @@ class RKMEImageSpecification(RegularStatsSpecification): """ if len(X.shape) != 4: - raise ValueError( - "X should be in shape of [N, C, {0:d}, {0:d}]. ".format(RKMEImageSpecification.IMAGE_WIDTH) - ) + raise ValueError("X should be in shape of [N, C, H, W]. ") if ( X.shape[2] != RKMEImageSpecification.IMAGE_WIDTH or X.shape[3] != RKMEImageSpecification.IMAGE_WIDTH