Collapse Unsloth shared API into train module

Author: Kovbo

src/art/loss.py

@@ -8,7 +8,7 @@
 from . import dev
 
 if TYPE_CHECKING:
-    from art.unsloth.service import TrainInputs
+    from art.preprocessing.inputs import TrainInputs
 
 
 class Loss(BaseModel):

src/art/unsloth/service.py

@@ -27,13 +27,13 @@
 from ..utils.get_model_step import get_step_from_dir
 from ..utils.output_dirs import get_step_checkpoint_dir
 from ..vllm import get_llm, get_worker, openai_server_task, run_on_workers
-from .shared import (
+from .train import (
     UnslothTrainContext,
     create_unsloth_train_context,
+    gc_and_empty_cuda_cache,
     run_unsloth_rl_training,
     run_unsloth_sft_training,
 )
-from .train import gc_and_empty_cuda_cache
 
 logger = logging.getLogger(__name__)