fix(trainer): import psutil to prevent NameError in _prepare_dataset (#3780)

* fix(trainer): import psutil to prevent NameError in _prepare_dataset

Fixes #3777

* Update rl.py

---------

Co-authored-by: Daniel Han <danielhanchen@gmail.com>

unsloth/models/rl.py

@@ -227,6 +227,7 @@ import numpy as np
 from contextlib import nullcontext
 from torch.nn import functional as F
 import inspect
+import psutil
 from transformers import DataCollatorForSeq2Seq, DataCollatorForLanguageModeling as TransformersDataCollatorForLanguageModeling
 from transformers.training_args import ParallelMode
 

unsloth/tokenizer_utils.py

@@ -25,6 +25,7 @@ import collections
 import numpy as np
 import gc
 import subprocess
+import psutil
 
 from unsloth_zoo.tokenizer_utils import (
     mean_of_trained_tokens,

unsloth/trainer.py

@@ -14,6 +14,7 @@
 
 import logging
 import os
+import psutil
 import warnings
 from dataclasses import dataclass, field
 from typing import Optional