<reponame>Jeevesh8/FireFighter-RL<gh_stars>0
import numpy as np
from networkx import grid_graph, to_numpy_matrix

config = {
    'adj_mat' : to_numpy_matrix(grid_graph(dim=(3,3)), dtype=np.bool),
    'initial_fire' : np.asarray([False, True, False]+[False]*6, dtype=np.bool),
    'burn_prob': 0.5,
    'n_defend': 2,
}
import numpy as np
from networkx import grid_graph, to_numpy_matrix

config = {
    'adj_mat' : to_numpy_matrix(grid_graph(dim=(3,3)), dtype=np.bool),
    'initial_fire' : np.asarray([False, True, False]+[False]*6, dtype=np.bool),
    'burn_prob': 0.5,
    'n_defend': 2,
}