Fix https://github.com/huggingface/transformers/issues/40292 (#40439)

* Fix https://github.com/huggingface/transformers/issues/40292

* Trigger tests

---------

Co-authored-by: Matt <rocketknight1@gmail.com>

src/transformers/integrations/integration_utils.py

@@ -41,7 +41,7 @@ from transformers.utils.import_utils import _is_package_available
 if os.getenv("WANDB_MODE") == "offline":
     print("⚙️  Running in WANDB offline mode")
 
-from .. import PreTrainedModel, TFPreTrainedModel, TrainingArguments
+from .. import PreTrainedModel, TrainingArguments
 from .. import __version__ as version
 from ..utils import (
     PushToHubMixin,
@@ -56,6 +56,9 @@ from ..utils import (
 
 logger = logging.get_logger(__name__)
 
+if is_tf_available():
+    from .. import TFPreTrainedModel
+
 if is_torch_available():
     import torch
     import torch.distributed as dist