ai-hub (#3)

Reviewed-on: #3
2026-06-03 21:06:06 +00:00
parent e9ada2612f
commit a3f3060e13
16 changed files with 1161 additions and 56 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -13,8 +13,10 @@ dependencies = [
    "boto3>=1.34,<1.42",
    "constructs>=10.0.0",
    "mlflow>=3.0",
+    "numpy>=1.26",
    "pydantic>=2.13.3",
    "pyyaml>=6.0.3",
+    "qai-hub>=0.49.0",
    "sagemaker-mlflow>=0.4.0",
 ]

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