From 622563f1730b15adb4059a3cb0c3608ef1d2de67 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 30 Apr 2025 08:45:04 -0700 Subject: [PATCH] Update synthetic.py --- unsloth/dataprep/synthetic.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/unsloth/dataprep/synthetic.py b/unsloth/dataprep/synthetic.py index 5cfd0f2a85..7119b903b4 100644 --- a/unsloth/dataprep/synthetic.py +++ b/unsloth/dataprep/synthetic.py @@ -13,10 +13,7 @@ # limitations under the License. __all__ = [ - "check_vllm_status", - "async_load_vllm", - "destroy_vllm", - "configure_synthetic_data_kit", + "SyntheticDataKit", ] import subprocess import time @@ -122,7 +119,7 @@ class SyntheticDataKit: if vllm_process is None: raise RuntimeError("Unsloth: vllm_process failed to load!") trial = 0 - while not check_vllm_status(): + while not self.check_vllm_status(): if trial >= 100: raise RuntimeError("Unsloth: vllm_process failed to load!") trial += 1