From 8276eceb8d919446fc7feeee4baf9bd2f2fa9848 Mon Sep 17 00:00:00 2001 From: Kyle Sayers Date: Fri, 14 Feb 2025 16:33:23 -0500 Subject: [PATCH] remove attribute Signed-off-by: Kyle Sayers --- src/llmcompressor/core/lifecycle.py | 3 --- .../llmcompressor/transformers/finetune/test_session_mixin.py | 1 - 2 files changed, 4 deletions(-) diff --git a/src/llmcompressor/core/lifecycle.py b/src/llmcompressor/core/lifecycle.py index 232d76b83..548341ff4 100644 --- a/src/llmcompressor/core/lifecycle.py +++ b/src/llmcompressor/core/lifecycle.py @@ -43,7 +43,6 @@ class CompressionLifecycle: modifiers: List[StageModifiers] = field(default_factory=list) event_lifecycle: Optional[EventLifecycle] = None - initialized_structure: bool = False initialized_: bool = False finalized: bool = False event_called: bool = False @@ -69,7 +68,6 @@ def reset(self): self.modifiers = [] self.event_lifecycle = None - self.initialized_structure = False self.initialized_ = False self.finalized = False self.event_called = False @@ -96,7 +94,6 @@ def pre_initialize_structure(self, **kwargs) -> List[Any]: if data is not None: mod_data.append(data) - self.initialized_structure = True applied_stage_names = [mod.unique_id for mod in self.modifiers if mod.applied] self.recipe_container.update_applied_stages(applied_stage_names) logger.info( diff --git a/tests/llmcompressor/transformers/finetune/test_session_mixin.py b/tests/llmcompressor/transformers/finetune/test_session_mixin.py index 93bd74cd1..f10c0ed51 100644 --- a/tests/llmcompressor/transformers/finetune/test_session_mixin.py +++ b/tests/llmcompressor/transformers/finetune/test_session_mixin.py @@ -64,5 +64,4 @@ def test_mixin_session_init(mixin_trainer): mixin_trainer.initialize_session(epoch=0.0, checkpoint=None) session = active_session() - assert not session.lifecycle.initialized_structure assert session.lifecycle.initialized_