Graphene library dependency for ModelTrainer

[sciarrilli]

Describe the bug
When importing ModelTrainer is errors because of a dependency on graphene.

To reproduce
A clear, step-by-step set of instructions to reproduce the bug.
The provided code need to be complete and runnable, if additional data is needed, please include them in the issue.

from sagemaker.modules.train import ModelTrainer

errors with

ModuleNotFoundError: No module named 'graphene'

Expected behavior
The import of ModelTrainer should succeed without an error.

Screenshots or logs
If applicable, add screenshots or logs to help explain your problem.

System information
A description of your system. Please provide:

• SageMaker Python SDK version: 2.243.0
• Framework name (eg. PyTorch) or algorithm (eg. KMeans):
• Framework version:
• Python version: 3.12.9
• CPU or GPU:
• Custom Docker image (Y/N):

Additional context
Add any other context about the problem here.