Method create_node

mcts_math/agents/react.py:63–67

(self, parent: Optional[Type[BaseNode]] = None)

        return cfg
    
    def create_node(self, parent: Optional[Type[BaseNode]] = None) -> Type[BaseNode]:
        return BaseNode(
            parent=parent, 
            additional_state_keys=self.REACT_NODE_KEYS,
        )

    def create_llm(self) -> Callable[[...], List[str]]:
        GPUS = os.environ.get('CUDA_VISIBLE_DEVICES', "0").split(',')