[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
ludwig-ai · Jan 22, 2024 · de19d8c · de19d8c
1 parent cc9bb6a
commit de19d8c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/ludwig/utils/batch_size_tuner.py b/ludwig/utils/batch_size_tuner.py
@@ -14,6 +14,7 @@
 
 TOTAL_STEPS = 5
 
+
 @DeveloperAPI
 class BatchSizeEvaluator(ABC):
     def select_best_batch_size(

diff --git a/tests/integration_tests/test_llm.py b/tests/integration_tests/test_llm.py
@@ -1254,6 +1254,7 @@ def test_llm_encoding(llm_encoder_config, adapter, quantization, tmpdir):
     model = LudwigModel(config)
     model.train(dataset=dataset_path, output_directory=str(tmpdir))
 
+
 def test_llm_batch_size_tuning():
     dataset = pd.DataFrame({"instruction": ["a"] * 100, "output": ["a"] * 100})
     config = yaml.safe_load(
@@ -1341,4 +1342,4 @@ def test_llm_used_tokens(tmpdir):
         "4": 0,
         "5": 204,
         "6": 0,
-    }
+    }