c9e82bb1ab323d4a79793e47dbdd9204302e36af,mlxtend/file_io/find.py,,find_filegroups,#,68

Before Change

                pass
    
    if validity_check:
        num = sorted([(len(v),k) for k,v in groups.items()])
        for i in num[1:]:
            if i[0] < num[0][0]:
                print("Warning, key "%s" has less values than "key" %s." % (i[1], num[0][1]))
                groups = {}

    return groups

After Change

    groups = {}
    for f in base:
        basename = os.path.splitext(os.path.basename(f))[0]
        basename = re.sub("\%s$" % rstrip, "", basename)
        groups[basename] = [f]
    
    //groups = {os.path.splitext(os.path.basename(f))[0].rstrip(rstrip):[f] for f in base}
    
    for idx,r in enumerate(rest):
        for f in r:
            basename, ext = os.path.splitext(os.path.basename(f))
            basename = re.sub("\%s$" % rstrip, "", basename)
            try:
                if extensions[idx+1] == "" or ext == extensions[idx+1]:
                    groups[basename].append(f)
            except KeyError:
                pass
    
    if validity_check:
        lens = [len(groups[k]) for k in groups.keys()]
        if len(set(lens)) > 1:
            raise ValueError("Warning, some keys have more/less values than others. Set validity_check=False to ignore this warning.")

    return groups

Instances