fix(pu): fix emplace_back

lzero/mcts/ctree/ctree_sampled_efficientzero/lib/cnode.cpp

@@ -380,12 +380,12 @@ namespace tree
             // After sorting, the first vector is the index, and the second vector is the probability value after perturbation sorted from large to small.
             for (size_t iter = 0; iter < disturbed_probs.size(); iter++)
             {
-//                #ifdef __APPLE__
-//                    disc_action_with_probs.__emplace_back(std::make_pair(iter, disturbed_probs[iter]));
-//                #else
-//                    disc_action_with_probs.emplace_back(std::make_pair(iter, disturbed_probs[iter]));
-//                #endif
-                  disc_action_with_probs.emplace_back(std::make_pair(iter, disturbed_probs[iter]));
+                #ifdef __APPLE__
+                    disc_action_with_probs.__emplace_back(std::make_pair(iter, disturbed_probs[iter]));
+                #else
+                    disc_action_with_probs.emplace_back(std::make_pair(iter, disturbed_probs[iter]));
+                #endif
+                // disc_action_with_probs.emplace_back(std::make_pair(iter, disturbed_probs[iter]));
             }
 
             std::sort(disc_action_with_probs.begin(), disc_action_with_probs.end(), cmp);

zoo/board_games/gomoku/envs/gomoku_env.py

@@ -1,32 +1,25 @@
 import copy
-import itertools
 import os
-import random
 import sys
 from functools import lru_cache
 from typing import List, Any
 
 import gym
 import imageio
+import matplotlib.patches as patches
+import matplotlib.pyplot as plt
 import numpy as np
 import pygame
 from ding.envs import BaseEnv, BaseEnvTimestep
 from ding.utils import ENV_REGISTRY
 from ditk import logging
 from easydict import EasyDict
-from matplotlib import pyplot as plt
-import matplotlib.pyplot as plt
-import matplotlib.patches as patches
-from zoo.board_games.gomoku.envs.legal_actions_cython import legal_actions_cython
 from zoo.board_games.gomoku.envs.get_done_winner_cython import get_done_winner_cython
+from zoo.board_games.gomoku.envs.legal_actions_cython import legal_actions_cython
 
 from zoo.board_games.alphabeta_pruning_bot import AlphaBetaPruningBot
-from zoo.board_games.connect4.envs.connect4_env import Connect4Env
 from zoo.board_games.gomoku.envs.gomoku_rule_bot_v0 import GomokuRuleBotV0
 from zoo.board_games.gomoku.envs.gomoku_rule_bot_v1 import GomokuRuleBotV1
-from zoo.board_games.gomoku.envs.utils import check_action_to_special_connect4_case1, \
-    check_action_to_special_connect4_case2, \
-    check_action_to_connect4
 
 
 @lru_cache(maxsize=512)

zoo/board_games/gomoku/envs/gomoku_rule_bot_v0.py

@@ -12,7 +12,7 @@ class GomokuRuleBotV0:
         forming a sequence of 4, forming a sequence of 3, forming a sequence of 2, and a random move.
     """
 
-    def __init__(self, env: Any, player: int, search_only_in_neighbor: bool = True) -> None:
+    def __init__(self, env: Any, player: int, search_only_in_neighbor: bool = False) -> None:
         """
         Overview:
             Initializes the bot with the game environment and the player it represents.

zoo/board_games/gomoku/envs/test_gomoku_rule_bot_v0.py

@@ -36,7 +36,7 @@ def test_naive(self):
                 print('action index of player 1 is:', action)
                 print('player 1: ' + env.action_to_string(action))
                 obs, reward, done, info = env.step(action)
-                env.render('image_realtime_mode')
+                # env.render('image_realtime_mode')
                 if done:
                     print('=' * 20)
                     if reward > 0:
@@ -52,7 +52,7 @@ def test_naive(self):
                 print('action index of player 2 is:', action)
                 print('player 2: ' + env.action_to_string(action))
                 obs, reward, done, info = env.step(action)
-                env.render('image_realtime_mode')
+                # env.render('image_realtime_mode')
                 if done:
                     print('=' * 20)
                     if reward > 0: