Cleaning up notebooks and adding mlflow dependencies (#1954)

Author: jam-jee

sagemaker-train/pyproject.toml

@@ -33,13 +33,15 @@ classifiers = [
   "Programming Language :: Python :: 3.12",
 ]
 dependencies = [
-  "sagemaker-core>=2.1.0",
-  "graphene>=3,<4",
-  "typing_extensions>=4.9.0",
-  "tblib>=1.7.0",
-  "PyYAML>=6.0,<7.0",
-  "paramiko>=2.11.0",
-  "jinja2>=3.0,<4.0"
+    "sagemaker-core>=2.1.0",
+    "graphene>=3,<4",
+    "typing_extensions>=4.9.0",
+    "tblib>=1.7.0",
+    "PyYAML>=6.0,<7.0",
+    "paramiko>=2.11.0",
+    "jinja2>=3.0,<4.0",
+    "sagemaker-mlflow>=0.0.1,<1.0.0",
+    "mlflow>=3.0.0,<4.0.0",
 ]
 
 [project.urls]