From 76518f7f1317a6cc6dc16af0aaa936c475dc416f Mon Sep 17 00:00:00 2001 From: shenwei41 Date: Fri, 21 Aug 2020 14:48:45 +0800 Subject: [PATCH] Modify the problem list --- mindspore/dataset/transforms/vision/validators.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/mindspore/dataset/transforms/vision/validators.py b/mindspore/dataset/transforms/vision/validators.py index f0eaba1793..9956c74a8c 100644 --- a/mindspore/dataset/transforms/vision/validators.py +++ b/mindspore/dataset/transforms/vision/validators.py @@ -519,8 +519,8 @@ def check_random_affine(method): check_degrees(degrees) if translate is not None: - if type_check(translate, (list, tuple), "translate"): - type_check_list(translate, (int, float), "translate") + type_check(translate, (list, tuple), "translate") + type_check_list(translate, (int, float), "translate") if len(translate) != 2: raise TypeError("translate should be a list or tuple of length 2.") for i, t in enumerate(translate): @@ -528,6 +528,7 @@ def check_random_affine(method): if scale is not None: type_check(scale, (tuple, list), "scale") + type_check_list(scale, (int, float), "scale") if len(scale) == 2: for i, s in enumerate(scale): check_positive(s, "scale[{}]".format(i)) @@ -541,6 +542,7 @@ def check_random_affine(method): if isinstance(shear, numbers.Number): check_positive(shear, "shear") else: + type_check_list(shear, (int, float), "shear") if len(shear) not in (2, 4): raise TypeError("shear must be of length 2 or 4.") if len(shear) == 2 and shear[0] > shear[1]: