search

OpenSourceEconomics / grmpy

Python package for the simulation and estimation of generalized Roy model
http://grmpy.readthedocs.io
MIT License
19 stars 5 forks source link

Missing Docstring #110

Closed peisenha closed 6 years ago

peisenha commented 6 years ago 
def refactor_results(dict_, file):
    pseudo = read(file)

    for key in ['TREATED', 'UNTREATED', 'COST', 'DIST']:
        if key == 'DIST':
            pseudo['DIST']['coeff'] = dict_['AUX']['x_internal'][-6:]
        else:
            pseudo[key]['coeff'] = dict_[key]['all'].tolist()
            del pseudo[key]['all']
    print_dict(pseudo, 'test')