run_tests.py 815 B

1234567891011121314151617181920212223
  1. import argparse
  2. import pytest
  3. from opendevin import config
  4. if __name__ == '__main__':
  5. """Main entry point of the script.
  6. This script runs pytest with specific arguments and configuration.
  7. Usage:
  8. python script_name.py [--OPENAI_API_KEY=<api_key>] [--model=<model_name>]
  9. """
  10. parser = argparse.ArgumentParser(description='This script runs pytest with specific arguments and configuration.')
  11. parser.add_argument('--OPENAI_API_KEY', type=str, required=True, help='Your OpenAI API key')
  12. parser.add_argument('--model', type=str, required=True, help='The model name to use')
  13. parser_args = parser.parse_args()
  14. config.config['OPENAI_API_KEY'] = parser_args.OPENAI_API_KEY
  15. args = ['-v', 'evaluation/regression/cases',f"-o model={parser_args.model}"]
  16. pytest.main(args)