feat: add __init__ method reference to LightningIRConfigMixin in model type mapping

RaykKretzschmar · RaykKretzschmar · commit 91c590900201 · 2026-06-12T11:27:42.000+02:00
diff --git a/lightning_ir/base/class_factory.py b/lightning_ir/base/class_factory.py
@@ -198,6 +198,7 @@ def from_backbone_class(self, BackboneClass: type[PretrainedConfig]) -> type[Lig
             f"{self.cc_lir_model_type(LightningIRConfigMixin)}{BackboneClass.__name__}",
             (LightningIRConfigMixin, BackboneClass),
             {
+                "__init__": LightningIRConfigMixin.__init__,
                 "model_type": self.MixinConfig.model_type,
                 "backbone_model_type": BackboneClass.model_type,
                 "mixin_config": self.MixinConfig,

Original file line number	Diff line number	Diff line change
`@@ -198,6 +198,7 @@ def from_backbone_class(self, BackboneClass: type[PretrainedConfig]) -> type[Lig`
`198`	`198`	`f"{self.cc_lir_model_type(LightningIRConfigMixin)}{BackboneClass.__name__}",`
`199`	`199`	`(LightningIRConfigMixin, BackboneClass),`
`200`	`200`	`{`
	`201`	`+ "__init__": LightningIRConfigMixin.__init__,`
`201`	`202`	`"model_type": self.MixinConfig.model_type,`
`202`	`203`	`"backbone_model_type": BackboneClass.model_type,`
`203`	`204`	`"mixin_config": self.MixinConfig,`