Skip to content

Commit 41308c4

Browse files
[tests] move recipe parsing test so it is run nightly (#1825)
SUMMARY: `text_recipe_parsing.py` was previously in `tests/llmcompressor/recipe`, but since it is decorated with `@requires_gpu` it was never being run. `llm-compressor-testing` only runs these tests if marked as `integration` tests and they are inside `tests/llmcompressor/transformers/(compression|finetune|obcq)` (see [here](https://github.com/neuralmagic/llm-compressor-testing/blob/main/.github/actions/test/action.yml#L28)). This moves the test into compression and marks it as an integration test so it will be included in tests. This is valid since it is calling into oneshot and running a compression flow on the model. TEST PLAN: - [x] Confirm with @dbarbuzzi that these tests are now being run in nightly. Tests should succeed now that #1810 is merged in --------- Signed-off-by: Brian Dellabetta <[email protected]>
1 parent 96ace59 commit 41308c4

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed
File renamed without changes.

tests/llmcompressor/recipe/test_recipe_parsing.py renamed to tests/llmcompressor/transformers/compression/test_recipe_parsing.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -67,13 +67,13 @@ def recipe_variants():
6767
scheme: W8A8
6868
"""
6969

70-
yaml_path = str(Path(__file__).parent / "recipe.yaml")
70+
yaml_path = str(Path(__file__).parent / "recipes" / "smoothquant_gptq_w8a8.yaml")
7171

7272
return [python_modifiers, yaml_string, yaml_path]
7373

7474

7575
@requires_gpu
76-
@pytest.mark.regression
76+
@pytest.mark.integration
7777
@pytest.mark.parametrize(
7878
"recipe",
7979
recipe_variants(),

0 commit comments

Comments
 (0)