Mlflow implementation (#2)

Reviewed-on: #2
2026-06-02 19:04:23 +00:00
parent 6ac9702dc5
commit e9ada2612f
13 changed files with 2287 additions and 38 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -12,8 +12,10 @@ dependencies = [
    "typer==0.25.0",
    "boto3>=1.34,<1.42",
    "constructs>=10.0.0",
+    "mlflow>=3.0",
    "pydantic>=2.13.3",
    "pyyaml>=6.0.3",
+    "sagemaker-mlflow>=0.4.0",
 ]

 [project.scripts]
@@ -25,6 +27,7 @@ packages = ["src"]
 [dependency-groups]
 dev = [
    "boto3-stubs[iam,s3,sagemaker]",
+    "pytest>=8.0",
    "pyright>=1.1.409",
    "types-PyYAML",
    "ruff>=0.4",