couldnt resist

Author: djsaunde

src/axolotl/datasets.py

@@ -174,7 +174,10 @@ def __iter__(self):
                             }
                         else:
                             LOG.warning(
-                                f"dropping batch due to tensor size mismatch input_ids: {input_ids.size()}, labels: {labels.size()}, attention_mask: {attention_mask.size()}"
+                                "Dropping batch due to tensor size mismatch "
+                                f"input_ids: {input_ids.size()}, "
+                                f"labels: {labels.size()}, "
+                                f"attention_mask: {attention_mask.size()}"
                             )
                     buffer = {
                         "input_ids": [],

src/axolotl/utils/data/wrappers.py

@@ -1,4 +1,4 @@
-"""data handling specific to SFT"""
+"""Data handling specific to SFT."""
 
 import logging
 from typing import Any, cast