From 263d6c3b86960aea15757feee052d2bd365cd63b Mon Sep 17 00:00:00 2001 From: Jovan Mitrevski Date: Tue, 30 Jun 2026 15:52:44 -0500 Subject: [PATCH] add tensorflow as the KERAS_BACKEND for qkerasV3 tests --- test/pytest/generate_ci_yaml.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/test/pytest/generate_ci_yaml.py b/test/pytest/generate_ci_yaml.py index d1c1e3009f..26775e5c0e 100644 --- a/test/pytest/generate_ci_yaml.py +++ b/test/pytest/generate_ci_yaml.py @@ -171,7 +171,11 @@ def generate_test_yaml(test_root='.'): name = 'qkerasV3' test_files = ' '.join([str(path.relative_to(test_root)) for path in batch_paths]) batch_need_example_model = int(any([qkeras3_need_examples[i] for i in batch_idxs])) - diff_yml = yaml.safe_load(template.format(name, '.pytest-qkeras-v3-only', test_files, batch_need_example_model)) + diff_yml = yaml.safe_load( + template_keras3_backend.format( + name, '.pytest-qkeras-v3-only', test_files, batch_need_example_model, 'tensorflow' + ) + ) yml.update(diff_yml) backend_specific_paths = [path for path in test_root.glob('**/test_*.py') if path.stem in KERAS3_BACKEND_SPECIFIC_LIST]