Skip to content

Util

arg_to_set(x)

convert None to empty set, iterable to set, or scalar to set with one item

Source code in src/notochord/util.py 
76
77
78
79
80
81
82
83
def arg_to_set(x):
    """convert None to empty set, iterable to set, or scalar to set with one item"""
    if x is None:
        return set()
    elif not hasattr(x, '__iter__'):
        return {x}
    else:
        return set(x)

deep_update(a, b)

in-place update a with contents of b, recursively for nested Mapping objects.

Source code in src/notochord/util.py 
65
66
67
68
69
70
71
72
73
def deep_update(a, b):
    """
    in-place update a with contents of b, recursively for nested Mapping objects.
    """
    for k in b:
        if k in a and isinstance(a[k], Mapping) and isinstance(b[k], Mapping):
            deep_update(a[k], b[k])
        else:
            a[k] = b[k]

gen_masks(n, dtype=torch.float)

yield the autoregressive mask matrices of all permuations of n items

Source code in src/notochord/util.py 
55
56
57
58
59
60
61
62
def gen_masks(n, dtype=torch.float):
    """yield the autoregressive mask matrices of all permuations of n items"""
    for perm in gen_perms(list(range(n))):
        m = torch.zeros(n,n,dtype=dtype)
        for idx,i in enumerate(perm):
            for j in perm[:idx]:
                m[j,i] = 1
        yield perm, m

gen_perms(a)

yield all permutations of the given list

Source code in src/notochord/util.py 
44
45
46
47
48
49
50
51
52
53
def gen_perms(a):
    """yield all permutations of the given list"""
    if len(a)==1:
        yield a
    else:
        # for each position
        for i in range(len(a)):
            # for permuations of remaining positions
            for p in gen_perms(a[:i]+a[i+1:]):  
                yield a[i:i+1]+p

get_class_defaults(cls)

get the default argument values of a class constructor

Source code in src/notochord/util.py 
28
29
30
31
32
33
34
35
36
37
38
39
40
41
def get_class_defaults(cls):
    """get the default argument values of a class constructor"""
    d = get_function_defaults(getattr(cls, '__init__'))
    # ignore `self` argument, insist on default values
    try:
        d.pop('self')
    except KeyError:
        raise ValueError("""
            no `self` argument found in class __init__
        """)
    assert [v is not inspect._empty for v in d.values()], """
            get_class_defaults should be used on constructors with keyword arguments only.
        """
    return d

get_function_defaults(fn)

get dict of name:default for a function's arguments

Source code in src/notochord/util.py 
23
24
25
26
def get_function_defaults(fn):
    """get dict of name:default for a function's arguments"""
    s = inspect.signature(fn)
    return {k:v.default for k,v in s.parameters.items()}