From 9db5072407741abd505606ed382e5f0ff96e670e Mon Sep 17 00:00:00 2001
From: Wing Lian <wing@axolotl.ai>
Date: Wed, 8 Jan 2025 22:30:13 -0500
Subject: [PATCH] make sure to use tensorboard to capture loss for checks

---
 tests/e2e/integrations/test_kd.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/e2e/integrations/test_kd.py b/tests/e2e/integrations/test_kd.py
index de6b1dcac..9e4103ecd 100644
--- a/tests/e2e/integrations/test_kd.py
+++ b/tests/e2e/integrations/test_kd.py
@@ -64,6 +64,7 @@ def min_cfg(temp_dir):
         "max_steps": 5,
         "output_dir": temp_dir,
         "save_safetensors": True,
+        "use_tensorboard": True,
     }