From 01fdae0adb703696310dc6f5874579c52c5dd37d Mon Sep 17 00:00:00 2001
From: Ihar Hrachyshka <ihar.hrachyshka@gmail.com>
Date: Tue, 20 May 2025 13:06:35 -0400
Subject: [PATCH] XXX

Signed-off-by: Ihar Hrachyshka <ihar.hrachyshka@gmail.com>
---
 tests/integration/post_training/test_post_training.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tests/integration/post_training/test_post_training.py b/tests/integration/post_training/test_post_training.py
index 9f0d887a7..960acf775 100644
--- a/tests/integration/post_training/test_post_training.py
+++ b/tests/integration/post_training/test_post_training.py
@@ -107,7 +107,8 @@ class TestPostTraining:
         logger.info(f"Starting training job with UUID: {self.job_uuid}")
 
         # train with HF trl SFTTrainer as the default
-        os.makedirs("~/.llama/checkpoints/", exist_ok=True)
+        checkpoint_dir = os.path.expanduser("/mnt/")
+        # os.makedirs(checkpoint_dir, exist_ok=True)
 
         started = datetime.now(timezone.utc)
         _ = llama_stack_client.post_training.supervised_fine_tune(
@@ -117,7 +118,7 @@ class TestPostTraining:
             training_config=training_config,
             hyperparam_search_config={},
             logger_config={},
-            checkpoint_dir="~/.llama/checkpoints/",
+            checkpoint_dir=checkpoint_dir,
         )
 
         while True: