Fix text-generation test

skavulya · May 7, 2024 · 7953cce · 7953cce
1 parent ed927ff
commit 7953cce
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_text_generation_example.py b/tests/test_text_generation_example.py
@@ -164,6 +164,7 @@ def _test_text_generation(
             env_variables["QUANT_CONFIG"] = os.path.join(
                 path_to_example_dir, "text-generation/quantization_config/maxabs_measure_include_outputs.json"
             )
+            command = [x for y in command for x in re.split(pattern, y) if x]
             subprocess.run(command, env=env_variables)
             env_variables["QUANT_CONFIG"] = os.path.join(
                 path_to_example_dir, "text-generation/quantization_config/maxabs_quant.json"