chore: lint

axolotl-ai-cloud · Jan 8, 2024 · 7f0ca82 · 7f0ca82
1 parent 28e505b
commit 7f0ca82
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git 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
@@ -1,3 +1,6 @@
+"""
+helper util to calculate dataset lengths
+"""
 import numpy as np