unilm/infoxlm/fairseq/scripts/compare_namespaces.py

#!/usr/bin/env python
"""Helper script to compare two argparse.Namespace objects."""

from argparse import Namespace  # noqa


def main():

    ns1 = eval(input('Namespace 1: '))
    ns2 = eval(input('Namespace 2: '))

    def keys(ns):
        ks = set()
        for k in dir(ns):
            if not k.startswith('_'):
                ks.add(k)
        return ks

    k1 = keys(ns1)
    k2 = keys(ns2)

    def print_keys(ks, ns1, ns2=None):
        for k in ks:
            if ns2 is None:
                print('{}\t{}'.format(k, getattr(ns1, k, None)))
            else:
                print('{}\t{}\t{}'.format(k, getattr(ns1, k, None), getattr(ns2, k, None)))

    print('Keys unique to namespace 1:')
    print_keys(k1 - k2, ns1)
    print()

    print('Keys unique to namespace 2:')
    print_keys(k2 - k1, ns2)
    print()

    print('Overlapping keys with different values:')
    ks = [k for k in k1 & k2 if getattr(ns1, k, 'None') != getattr(ns2, k, 'None')]
    print_keys(ks, ns1, ns2)
    print()


if __name__ == '__main__':
    main()