move import of llmcompressor to reset session inside test

This commit is contained in:
Wing Lian
2025-04-30 18:10:44 -04:00
parent 0ed4b4c310
commit 6affbb1f85

View File

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