Python register_script_config Examples

Example #1

File: parlai_test_script.py Project: shirleypatterson/Mephisto

    num_turns: int = field(
        default=3,
        metadata={
            'help': "Number of turns before a conversation is complete",
        },
    )
    turn_timeout: int = field(
        default=300,
        metadata={
            'help': 
                "Maximum response time before kicking "
                "a worker out, default 300 seconds",
        },
    )

register_script_config(name='scriptconfig', module=TestScriptConfig)


@hydra.main(config_name='scriptconfig')
def main(cfg: DictConfig) -> None:
    db, cfg = load_db_and_process_config(cfg)

    world_opt = {
        "num_turns": cfg.num_turns, 
        "turn_timeout": cfg.turn_timeout,
    }

    custom_bundle_path = cfg.mephisto.blueprint.get('custom_source_bundle', None)
    if custom_bundle_path is not None:
        assert os.path.exists(custom_bundle_path), (
            "Must build the custom bundle with `npm install; npm run dev` from within "

Example #2

File: parlai_test_script.py Project: mikejqzhang/Mephisto

    task_dir: str = TASK_DIRECTORY
    num_turns: int = field(
        default=3,
        metadata={"help": "Number of turns before a conversation is complete"},
    )
    turn_timeout: int = field(
        default=300,
        metadata={
            "help":
            "Maximum response time before kicking "
            "a worker out, default 300 seconds"
        },
    )


register_script_config(name="scriptconfig", module=TestScriptConfig)


@hydra.main(config_name="scriptconfig")
def main(cfg: DictConfig) -> None:
    db, cfg = load_db_and_process_config(cfg)

    world_opt = {"num_turns": cfg.num_turns, "turn_timeout": cfg.turn_timeout}

    custom_bundle_path = cfg.mephisto.blueprint.get("custom_source_bundle",
                                                    None)
    if custom_bundle_path is not None:
        assert os.path.exists(custom_bundle_path), (
            "Must build the custom bundle with `npm install; npm run dev` from within "
            f"the {TASK_DIRECTORY}/webapp directory in order to demo a custom bundle "
        )