72f89ffbeaad419cad9abc7c2d6a68bc1304396c,torchsample/modules/_utils.py,,validate_optimizer_input,#Any#,34

Before Change

def validate_optimizer_input(optimizer):
    if isinstance(optimizer, str):
        try:
            optimizer = eval("optim.%s" % optimizer)
        except:
            raise ValueError("Invalid optimizer string input - must match torch function exactly!")
        return optimizer
    elif hasattr(optimizer, "step") and hasattr(optimizer, "zero_grad"):
        return optimizer
    else:
        raise ValueError("Invalid optimizer input")

After Change

def validate_optimizer_input(optimizer):
    dir_optim = dir(optim)
    opts = [o.lower() for o in dir_optim]
    if isinstance(optimizer, str):
        try:
            str_idx = opts.index(optimizer.lower())    
        except:
            raise ValueError("Invalid optimizer string input - must match pytorch function.")
        return getattr(optim, dir_optim[str_idx])
    elif hasattr(optimizer, "step") and hasattr(optimizer, "zero_grad"):
        return optimizer
    else:
        raise ValueError("Invalid optimizer input")

Instances