def save_params_in_scopes()

in coinrun/main_utils.py [0:0]

• 15 lines of code
• 4 McCabe index (conditional complexity)

def save_params_in_scopes(sess, scopes, filename, base_dict=None):
    data_dict = {}

    if base_dict is not None:
        data_dict.update(base_dict)

    save_path = file_to_path(filename)

    data_dict['args'] = Config.get_args_dict()
    param_dict = {}

    for scope in scopes:
        params = tf.trainable_variables(scope)

        if len(params) > 0:
            print('saving scope', scope, filename)
            ps = sess.run(params)

            param_dict[scope] = ps
        
    data_dict['params'] = param_dict
    joblib.dump(data_dict, save_path)