def setup_method(self): EvalbBracketingScorer.compile_evalb() super().setup_method() self.set_up_model( FIXTURES_ROOT / "syntax" / "constituency_parser" / "constituency_parser.json", FIXTURES_ROOT / "syntax" / "example_ptb.trees", )
def setUp(self): EvalbBracketingScorer.compile_evalb() super(SpanConstituencyParserTest, self).setUp() self.set_up_model( self.FIXTURES_ROOT / u"constituency_parser" / u"constituency_parser.json", self.FIXTURES_ROOT / u"data" / u"example_ptb.trees")
def setup_method(self): super().setup_method() EvalbBracketingScorer.compile_evalb()
logger.info("Changing directory to %s", module_root) os.chdir(module_root) if len(sys.argv) >= 2 and sys.argv[1].lower() == "gpu": train_fixture_gpu("allennlp/tests/fixtures/srl/") else: models = [ 'biaffine_dependency_parser', 'bidaf', 'dialog_qa', 'constituency_parser', 'coref', 'decomposable_attention', 'encoder_decoder/composed_seq2seq', 'encoder_decoder/simple_seq2seq', 'encoder_decoder/copynet_seq2seq', 'semantic_parsing/nlvr_coverage_semantic_parser', 'semantic_parsing/nlvr_direct_semantic_parser', 'semantic_parsing/wikitables', 'semantic_parsing/quarel', 'semantic_parsing/quarel/zeroshot', 'semantic_parsing/quarel/tagger', 'semantic_parsing/atis', 'srl', ] for model in models: if model == 'constituency_parser': EvalbBracketingScorer.compile_evalb() train_fixture(f"allennlp/tests/fixtures/{model}/") logger.info("Changing directory back to %s", initial_working_dir) os.chdir(initial_working_dir)
def setUp(self): super().setUp() EvalbBracketingScorer.compile_evalb()
def setUp(self): EvalbBracketingScorer.compile_evalb() super(SpanConstituencyParserTest, self).setUp() self.set_up_model(self.FIXTURES_ROOT / "constituency_parser" / "constituency_parser.json", self.FIXTURES_ROOT / "data" / "example_ptb.trees")
def setUp(self): EvalbBracketingScorer.compile_evalb() super(SpanConstituencyParserTest, self).setUp() self.set_up_model("tests/fixtures/constituency_parser/constituency_parser.json", "tests/fixtures/data/example_ptb.trees")
if __name__ == "__main__": initial_working_dir = os.getcwd() module_root = _get_module_root().parent logger.info("Changing directory to %s", module_root) os.chdir(module_root) if len(sys.argv) >= 2 and sys.argv[1].lower() == "gpu": train_fixture_gpu("allennlp/tests/fixtures/srl/") else: models = [ 'biaffine_dependency_parser', 'bidaf', 'dialog_qa', 'constituency_parser', 'coref', 'decomposable_attention', 'encoder_decoder/simple_seq2seq', 'semantic_parsing/nlvr_coverage_semantic_parser', 'semantic_parsing/nlvr_direct_semantic_parser', 'semantic_parsing/wikitables', 'semantic_parsing/atis', 'srl', ] for model in models: if model == 'constituency_parser': EvalbBracketingScorer.compile_evalb() train_fixture(f"allennlp/tests/fixtures/{model}/") logger.info("Changing directory back to %s", initial_working_dir) os.chdir(initial_working_dir)