tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

move import of llmcompressor to reset session inside test

Browse Source
This commit is contained in:
Wing Lian
2025-04-30 18:10:44 -04:00
parent 0ed4b4c310
commit 6affbb1f85
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/e2e/integrations/test_llm_compressor.py
View File

@@ -5,7 +5,6 @@ E2E smoke tests for LLMCompressorPlugin integration
from pathlib import Path from pathlib import Path
import pytest import pytest
from llmcompressor import active_session
from axolotl.cli.args import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
@@ -41,6 +40,8 @@ class TestLLMCompressorIntegration:
def test_llmcompressor_plugin( def test_llmcompressor_plugin(
self, temp_dir, base_model: str, save_compressed: bool self, temp_dir, base_model: str, save_compressed: bool
): ):
from llmcompressor import active_session
# core cfg # core cfg
cfg = DictDefault( cfg = DictDefault(
{ {