# Path: util/logger.py
# class _MyFormatter(logging.Formatter):
#     def format(self, record):
# def set_file_handler(path=None, prefix='', time_str=''):
# _LOGGING_METHOD = ['info', 'warning', 'error', 'critical',
#                    'warn', 'exception', 'debug']
# 
# Path: graph_util/structure_mapper.py
# _BASE_DIR = init_path.get_base_dir()
# def map_output(transfer_env, i_value, added_constant, gnn_option_list):
# def map_input(transfer_env, i_value, added_constant, gnn_option_list):
# def map_transfer_env_running_mean(ienv, oenv, running_mean_info,
#                                   observation_size,
#                                   gnn_node_option, root_connection_option,
#                                   gnn_output_option, gnn_embedding_option):


import init_path
import numpy as np
import os
from util import logger
from graph_util import structure_mapper
                if transfer_env == 'Nothing2Nothing' or mlp_raw_transfer == 2:
                    # for the baseline, no one is preloading!
                    logger.warning('Skipping variable: {}'.format(var.name))
                else:
                    # load the policy logstd, happens for the gnn, mlpr and mlpt
                    if 'policy_logstd' in var.name:
                        if 'fix' in logstd_option:
                            single_value = float(logstd_option.split('_')[1])
                            assign_op = var.assign(
                                np.ones(var.get_shape().as_list()) *
                                single_value
                            )
                        else:
                            if '_' in logstd_option:
                                added_constant = float(
                                    logstd_option.split('_')[1]
                                )
                            else:
                                added_constant = 0.0

                            new_value = structure_mapper.map_output(
                                transfer_env,
                                val,
                                added_constant,
                                gnn_option_list
                            )
                            assign_op = var.assign(new_value)

                        sess.run(assign_op)  # or `assign_op.op.run()`
                    elif 'policy_0/w:0' in var.name:

# Path: util/logger.py
# class _MyFormatter(logging.Formatter):
#     def format(self, record):
# def set_file_handler(path=None, prefix='', time_str=''):
# _LOGGING_METHOD = ['info', 'warning', 'error', 'critical',
#                    'warn', 'exception', 'debug']
# 
# Path: graph_util/structure_mapper.py
# _BASE_DIR = init_path.get_base_dir()
# def map_output(transfer_env, i_value, added_constant, gnn_option_list):
# def map_input(transfer_env, i_value, added_constant, gnn_option_list):
# def map_transfer_env_running_mean(ienv, oenv, running_mean_info,
#                                   observation_size,
#                                   gnn_node_option, root_connection_option,
#                                   gnn_output_option, gnn_embedding_option):


# Path: util/model_saver.py
import init_path
import numpy as np
import os
from util import logger
from graph_util import structure_mapper

                if transfer_env == 'Nothing2Nothing' or mlp_raw_transfer == 2:
                    # for the baseline, no one is preloading!
                    logger.warning('Skipping variable: {}'.format(var.name))
                else:
                    # load the policy logstd, happens for the gnn, mlpr and mlpt
                    if 'policy_logstd' in var.name:
                        if 'fix' in logstd_option:
                            single_value = float(logstd_option.split('_')[1])
                            assign_op = var.assign(
                                np.ones(var.get_shape().as_list()) *
                                single_value
                            )
                        else:
                            if '_' in logstd_option:
                                added_constant = float(
                                    logstd_option.split('_')[1]
                                )
                            else:
                                added_constant = 0.0

                            new_value = structure_mapper.map_output(
                                transfer_env,
                                val,
                                added_constant,
                                gnn_option_list
                            )
                            assign_op = var.assign(new_value)

                        sess.run(assign_op)  # or `assign_op.op.run()`
                    elif 'policy_0/w:0' in var.name:

                        new_value = structure_mapper.map_input(