From 9be1495a2f68fe04d796bb31a9d13432ea4f37fb Mon Sep 17 00:00:00 2001 From: shihy Date: Sun, 12 Nov 2023 16:29:38 +0800 Subject: [PATCH] [MNT] modify error msg --- learnware/specification/regular/image/rkme.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) 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