inital ai hub implementation

2026-06-01 15:14:10 -04:00
parent 57a8a0a9c4
commit d3ebd2cc5f
12 changed files with 833 additions and 15 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",
 ]

@@ -28,6 +30,7 @@ packages = ["src"]
 dev = [
    "boto3-stubs[iam,s3,sagemaker]",
    "pytest>=8.0",
+    "pytest-mock>=3.12",
    "pyright>=1.1.409",
    "types-PyYAML",
    "ruff>=0.4",