ast_toolbox.mcts.AdaptiveStressTestingRandomSeed module

class ast_toolbox.mcts.AdaptiveStressTestingRandomSeed.ASTRSAction(action, env)

Bases: object

The AST action containing the random seed.

Parameters:

action – The random seed. env : ast_toolbox.envs.go_explore_ast_env.GoExploreASTEnv

The environment.

get()

Get the true action.

Returns:action – The true actions used in the env.

class ast_toolbox.mcts.AdaptiveStressTestingRandomSeed.AdaptiveStressTestRS(**kwargs)

Bases: ast_toolbox.mcts.AdaptiveStressTesting.AdaptiveStressTest

The AST wrapper for MCTS using random seeds as actions.

Parameters:kwargs – Keyword arguments passed to ast_toolbox.mcts.AdaptiveStressTesting.AdaptiveStressTest

explore_action(s, tree)

Randomly sample an action for the exploration.

Returns:action (ast_toolbox.mcts.AdaptiveStressTestingRandomSeed.ASTRSAction) – The sampled action.

random_action()

Randomly sample an action for the rollout.

Returns:action (ast_toolbox.mcts.AdaptiveStressTestingRandomSeed.ASTRSAction) – The sampled action.

reset_rsg()

Reset the random seed generator.