diff --git a/src/axolotl/utils/samplers/__init__.py b/src/axolotl/utils/samplers/__init__.py
index a3a521751..96e00a5d2 100644
--- a/src/axolotl/utils/samplers/__init__.py
+++ b/src/axolotl/utils/samplers/__init__.py
@@ -2,4 +2,4 @@
 axolotl samplers module
 """
 from .multipack import MultipackBatchSampler  # noqa: F401
-from .utils import get_dataset_lengths
+from .utils import get_dataset_lengths  # noqa: F401
diff --git a/src/axolotl/utils/samplers/utils.py b/src/axolotl/utils/samplers/utils.py
index 0ea1ce6c9..926c7386a 100755
--- a/src/axolotl/utils/samplers/utils.py
+++ b/src/axolotl/utils/samplers/utils.py
@@ -1,3 +1,6 @@
+"""
+helper util to calculate dataset lengths
+"""
 import numpy as np