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)