def new_method(self, *args, **kwargs):
[degrees, resample, expand, center, fill_value], _ = parse_user_args(method, *args, **kwargs)
check_degrees(degrees)
if resample is not None:
type_check(resample, (Inter,), "resample")
if expand is not None:
type_check(expand, (bool,), "expand")
if center is not None:
check_2tuple(center, "center")
if fill_value is not None:
check_fill_value(fill_value)
return method(self, *args, **kwargs)
def new_method(self, *args, **kwargs):
[policy], _ = parse_user_args(method, *args, **kwargs)
type_check(policy, (list,), "policy")
if not policy:
raise ValueError("policy can not be empty.")
for sub_ind, sub in enumerate(policy):
type_check(sub, (list,), "policy[{0}]".format([sub_ind]))
if not sub:
raise ValueError("policy[{0}] can not be empty.".format(sub_ind))
for op_ind, tp in enumerate(sub):
check_2tuple(tp, "policy[{0}][{1}]".format(sub_ind, op_ind))
check_tensor_op(tp[0], "op of (op, prob) in policy[{0}][{1}]".format(sub_ind, op_ind))
check_value(tp[1], (0, 1), "prob of (op, prob) policy[{0}][{1}]".format(sub_ind, op_ind))
return method(self, *args, **kwargs)
