linter

Author: Borda

src/litmodels/integrations/__init__.py

@@ -1,4 +1,5 @@
 """Integrations with training frameworks like PyTorch Lightning, TensorFlow, and others."""
 
 from litmodels.integrations.checkpoints import LitModelCheckpoint
-from litmodels.integrations.duplicate import duplicate_hf_model
+
+__all__ = ["LitModelCheckpoint"]

tests/integrations/test_lightning.py

@@ -1,8 +1,8 @@
 import re
 from unittest import mock
 
-from litmodels.integrations.imports import _LIGHTNING_AVAILABLE, _PYTORCHLIGHTNING_AVAILABLE
 from litmodels.integrations.checkpoints import LitModelCheckpoint
+from litmodels.integrations.imports import _LIGHTNING_AVAILABLE, _PYTORCHLIGHTNING_AVAILABLE
 
 if _LIGHTNING_AVAILABLE:
     from lightning import Trainer