From 13dae882508bf13b3e479df00b73336e9400e8e9 Mon Sep 17 00:00:00 2001 From: whimo Date: Fri, 3 May 2024 22:55:59 +0400 Subject: [PATCH] Minor refactoring --- tests/run_integration_tests.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/tests/run_integration_tests.py b/tests/run_integration_tests.py index 63330e97..3db18ac0 100644 --- a/tests/run_integration_tests.py +++ b/tests/run_integration_tests.py @@ -20,18 +20,17 @@ from examples.delegation_crewai import main as delegation_crewai_main from examples.single_llama_index import main as single_llama_index_main -from examples.single_openai_tools_react import main as single_openai_tools_react_main -CACHE_DIR = Path(__file__).parent / "itest_cache" -GOLDEN_DIR = Path(__file__).parent / "itest_golden_data" - INTEGRATION_TESTS = { "single_llama_index": single_llama_index_main, "delegation_crewai": delegation_crewai_main, # "single_openai_tools_react": single_openai_tools_react_main, TODO: enable this test } +DEFAULT_CACHE_DIR = Path(__file__).parent / "itest_cache" +DEFAULT_GOLDEN_DIR = Path(__file__).parent / "itest_golden_data" + def get_args_parser(): """Argument parser""" @@ -45,9 +44,9 @@ def get_args_parser(): help="Name of the test to run (leave empty to run all tests)", default=None, ) - parser.add_argument("--cache-dir", type=str, help="Cache directory", default=CACHE_DIR) + parser.add_argument("--cache-dir", type=str, help="Cache directory", default=DEFAULT_CACHE_DIR) parser.add_argument( - "--golden-dir", type=str, help="Reference data directory", default=GOLDEN_DIR + "--golden-dir", type=str, help="Reference data directory", default=DEFAULT_GOLDEN_DIR ) parser.add_argument( "--update-golden",