# 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']


import numpy as np
import init_path
from util import logger
from six.moves import xrange
#!/usr/bin/env python2
# -----------------------------------------------------------------------------
#   @brief:
#       In this function, we change the data from [batch_size, ob_dim] into
#       [batch_size * num_node, hidden_dim]
#   @author:
#       Tingwu Wang, Jul. 13th, 2017
# -----------------------------------------------------------------------------



_ABS_BASE_PATH = init_path.get_abs_base_dir()


def construct_graph_input_feeddict(node_info,
                                   obs_n,
                                   receive_idx,
                                   send_idx,

# 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/graph_data_util.py
import numpy as np
import init_path
from util import logger
from six.moves import xrange

#!/usr/bin/env python2
# -----------------------------------------------------------------------------
#   @brief:
#       In this function, we change the data from [batch_size, ob_dim] into
#       [batch_size * num_node, hidden_dim]
#   @author:
#       Tingwu Wang, Jul. 13th, 2017
# -----------------------------------------------------------------------------



_ABS_BASE_PATH = init_path.get_abs_base_dir()


def construct_graph_input_feeddict(node_info,
                                   obs_n,
                                   receive_idx,
                                   send_idx,

                                   node_type_idx,