def setUp(self):
opts = MockOpt()
opts.ground_truth_data_dir = GROUND_TRUTH_DATA_DIR
self.dialogue_manager = DialogueManager(
memory=None,
dialogue_object_classes={},
opts=opts,
low_level_interpreter_data={},
)
def setUp(self):
opts = MockOpt()
opts.nsp_models_dir = TTAD_MODEL_DIR
opts.nsp_data_dir = TTAD_BERT_DATA_DIR
super().setUp(agent_opts=opts)
self.cube_right = self.add_object(
droidlet.lowlevel.minecraft.shapes.cube(bid=(42, 0)), (9, 63, 4))
self.cube_left = self.add_object(
droidlet.lowlevel.minecraft.shapes.cube(), (9, 63, 10))
self.set_looking_at(list(self.cube_right.blocks.keys())[0])
def setUp(self):
opts = MockOpt()
opts.nsp_data_dir = TTAD_BERT_DATA_DIR
opts.ground_truth_data_dir = GROUND_TRUTH_DATA_DIR
opts.nsp_models_dir = TTAD_MODEL_DIR
opts.no_ground_truth = False
self.chat_parser = NSPQuerier(opts=opts)
def __init__(self, *args, **kwargs):
super(TestDialogueManager, self).__init__(*args, **kwargs)
opts = MockOpt()
opts.nsp_data_dir = TTAD_BERT_DATA_DIR
opts.ground_truth_data_dir = GROUND_TRUTH_DATA_DIR
opts.nsp_models_dir = TTAD_MODEL_DIR
opts.no_ground_truth = False
self.agent = FakeAgent(opts)
def default_scenario(args, command_text, verifier):
scenario = {}
scenario["command_text"] = command_text
J = build_shape_scene(args)
world_spec = {}
world_spec["coord_shift"] = J["offset"]
def init_blocks(world):
for b in J["schematic_for_cuberite"]:
world.place_block(((b["x"], b["y"], b["z"]), (b["id"], b["meta"])))
world_spec["ground_generator"] = init_blocks
speaker_pos = np.add(J["avatarInfo"]["pos"], J["offset"]).tolist()
speaker = FakePlayer(
Player(42, PLAYER_NAME, Pos(*speaker_pos), Look(*J["avatarInfo"]["look"]), Item(0, 0)),
active=False,
opts=None,
)
world_spec["players"] = [speaker]
world_spec["mobs"] = []
world_spec["item_stacks"] = []
agent_pos = np.add(J["agentInfo"]["pos"], J["offset"]).tolist()
# FIXME normalize this interface
world_spec["agent"] = {"pos": agent_pos}
scenario["world_spec"] = world_spec
world_opts = MockOpt()
world_opts.sl = SL
scenario["world_opts"] = world_opts
agent_opts = MockOpt()
agent_opts.nsp_models_dir = TTAD_MODEL_DIR
agent_opts.nsp_data_dir = TTAD_BERT_DATA_DIR
agent_opts.e2e_mode = True
scenario["agent_opts"] = agent_opts
scenario["max_execution_steps"] = 100
scenario["verifier"] = verifier
return scenario
def __init__(self, world, opts=None, do_heuristic_perception=False, prebuilt_perception=None):
self.mark_airtouching_blocks = do_heuristic_perception
self.head_height = HEAD_HEIGHT
self.world = world
self.chat_count = 0
# use these to not have to re-init models if running many tests:
self.prebuilt_perception = prebuilt_perception
if not opts:
opts = MockOpt()
self.e2e_mode = getattr(opts, "e2e_mode", False)
self.low_level_data = {
"mobs": SPAWN_OBJECTS,
"mob_property_data": craftassist_specs.get_mob_property_data(),
"schematics": craftassist_specs.get_schematics(),
"block_data": craftassist_specs.get_block_data(),
"block_property_data": craftassist_specs.get_block_property_data(),
"color_data": craftassist_specs.get_colour_data(),
"boring_blocks": BORING_BLOCKS,
"passable_blocks": PASSABLE_BLOCKS,
"fill_idmeta": fill_idmeta,
"color_bid_map": COLOR_BID_MAP,
}
super(FakeAgent, self).__init__(opts)
self.do_heuristic_perception = do_heuristic_perception
self.no_default_behavior = True
self.last_task_memid = None
pos = (0, 63, 0)
if hasattr(self.world, "agent_data"):
pos = self.world.agent_data["pos"]
self.pos = np.array(pos, dtype="int")
self.logical_form = None
self.world_interaction_occurred = False
self.world_interaction_occurred = False
self._held_item: IDM = (0, 0)
self._look_vec = (1, 0, 0)
self._changed_blocks: List[Block] = []
self._outgoing_chats: List[str] = []
CraftAssistAgent.add_self_memory_node(self)
def setUp(self):
opts = MockOpt()
opts.nsp_data_dir = TTAD_BERT_DATA_DIR
opts.ground_truth_data_dir = GROUND_TRUTH_DATA_DIR
opts.nsp_models_dir = TTAD_MODEL_DIR
opts.no_ground_truth = False
self.chat_parser = NSPQuerier(opts=opts)
self.ground_truth_actions = {}
print("fetching data from ground truth, from directory: %r" %
(opts.ground_truth_data_dir))
if not opts.no_ground_truth:
if os.path.isdir(opts.ground_truth_data_dir):
dataset = opts.ground_truth_data_dir + "datasets/high_pri_commands.txt"
with open(dataset) as f:
for line in f.readlines():
text, logical_form = line.strip().split("|")
clean_text = text.strip('"').lower()
self.ground_truth_actions[clean_text] = json.loads(
logical_form)
self.ground_truth_actions.update(GROUND_TRUTH_PARSES)
def test_init_agent(self):
opts = MockOpt()
CraftAssistAgent(opts)
def setUp(self):
opts = MockOpt()
opts.nsp_data_dir = TTAD_BERT_DATA_DIR
opts.nsp_models_dir = TTAD_BERT_MODEL_DIR
opts.no_ground_truth = False
super().setUp(agent_opts=opts)
def setUp(self):
opts = MockOpt()
opts.ground_truth_data_dir = GROUND_TRUTH_DATA_DIR
opts.no_ground_truth = False
super().setUp(agent_opts=opts)