diff --git a/douzero/env/game.py b/douzero/env/game.py index 1b716f7..99c00d5 100644 --- a/douzero/env/game.py +++ b/douzero/env/game.py @@ -258,27 +258,27 @@ class GameEnv(object): elif rival_move_type == md.TYPE_4_BOMB: all_moves = mg.gen_type_4_bomb(4) moves = ms.filter_type_4_bomb(all_moves, rival_move) - all_moves += mg.gen_type_4_bomb(5) + mg.gen_type_4_bomb(6) + mg.gen_type_4_bomb(7) + mg.gen_type_4_bomb(8) + mg.gen_type_5_king_bomb() + moves += mg.gen_type_4_bomb(5) + mg.gen_type_4_bomb(6) + mg.gen_type_4_bomb(7) + mg.gen_type_4_bomb(8) + mg.gen_type_5_king_bomb() elif rival_move_type == md.TYPE_4_BOMB5: all_moves = mg.gen_type_4_bomb(5) moves = ms.filter_type_4_bomb(all_moves, rival_move) - all_moves += mg.gen_type_4_bomb(6) + mg.gen_type_4_bomb(7) + mg.gen_type_4_bomb(8) + mg.gen_type_5_king_bomb() + moves += mg.gen_type_4_bomb(6) + mg.gen_type_4_bomb(7) + mg.gen_type_4_bomb(8) + mg.gen_type_5_king_bomb() elif rival_move_type == md.TYPE_4_BOMB6: all_moves = mg.gen_type_4_bomb(6) moves = ms.filter_type_4_bomb(all_moves, rival_move) - all_moves += mg.gen_type_4_bomb(7) + mg.gen_type_4_bomb(8) + mg.gen_type_5_king_bomb() + moves += mg.gen_type_4_bomb(7) + mg.gen_type_4_bomb(8) + mg.gen_type_5_king_bomb() elif rival_move_type == md.TYPE_4_BOMB7: all_moves = mg.gen_type_4_bomb(7) moves = ms.filter_type_4_bomb(all_moves, rival_move) - all_moves += mg.gen_type_4_bomb(8) + mg.gen_type_5_king_bomb() + moves += mg.gen_type_4_bomb(8) + mg.gen_type_5_king_bomb() elif rival_move_type == md.TYPE_4_BOMB8: all_moves = mg.gen_type_4_bomb(8) moves = ms.filter_type_4_bomb(all_moves, rival_move) - all_moves += mg.gen_type_5_king_bomb() + moves += mg.gen_type_5_king_bomb() elif rival_move_type == md.TYPE_5_KING_BOMB: moves = []