run_tests.py 867 B

123456789101112131415161718192021222324252627282930
  1. import argparse
  2. import pytest
  3. from opendevin.config 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(
  11. description='This script runs pytest with specific arguments and configuration.'
  12. )
  13. parser.add_argument(
  14. '--OPENAI_API_KEY', type=str, required=True, help='Your OpenAI API key'
  15. )
  16. parser.add_argument(
  17. '--model', type=str, required=True, help='The model name to use'
  18. )
  19. parser_args = parser.parse_args()
  20. config.config['OPENAI_API_KEY'] = parser_args.OPENAI_API_KEY
  21. args = ['-v', 'evaluation/regression/cases', f'-o model={parser_args.model}']
  22. pytest.main(args)