fix: Move sagemaker-mlflow to extras

ryansteakley · ryansteakley · commit 12ba2ec4ad89 · 2024-10-17T19:11:23.000Z
diff --git a/hatch_build.py b/hatch_build.py
@@ -20,7 +20,7 @@ def read_feature_deps(feature):
 
     optional_dependencies = {"all": []}
 
-    for feature in ("feature-processor", "huggingface", "local", "scipy"):
+    for feature in ("feature-processor", "huggingface", "local", "scipy", "sagemaker-mlflow"):
         dependencies = read_feature_deps(feature)
         optional_dependencies[feature] = dependencies
         optional_dependencies["all"].extend(dependencies)
diff --git a/pyproject.toml b/pyproject.toml
@@ -48,7 +48,6 @@ dependencies = [
   "PyYAML~=6.0",
   "requests",
   "sagemaker-core>=1.0.0,<2.0.0",
-  "sagemaker-mlflow",
   "schema",
   "smdebug_rulesconfig==1.0.1",
   "tblib>=1.7.0,<4",
diff --git a/requirements/extras/sagemaker-mlflow_requirements.txt b/requirements/extras/sagemaker-mlflow_requirements.txt
@@ -0,0 +1 @@
+sagemaker-mlflow
