Update conda source (#6)
diff --git a/Project-AutoML/automl/mod/mod_autosklearn.py b/Project-AutoML/automl/mod/mod_autosklearn.py
index e13b715..40ebf45 100644
--- a/Project-AutoML/automl/mod/mod_autosklearn.py
+++ b/Project-AutoML/automl/mod/mod_autosklearn.py
@@ -31,21 +31,23 @@
model_path = "autosklearn.pkl"
conda_env = {
- "channels": ["defaults", "conda-forge"],
"dependencies": [
"python=3.8.2",
+ "pip",
{
"pip": [
"mlflow",
+ "click==8.0.",
"scikit-learn==0.24.2",
"boto3==1.22.2",
- "pandas==1.3.5",
+ "pandas>=1.0.0",
"setuptools<59.6.0",
"auto-sklearn==0.14.6",
+ "flaml==1.0.1"
],
},
],
- "name": "mlflow-env",
+ "name": "MLflow-AutoML",
}
@staticmethod
diff --git a/Project-AutoML/automl/mod/mod_flaml.py b/Project-AutoML/automl/mod/mod_flaml.py
index d371b1f..e699ee2 100644
--- a/Project-AutoML/automl/mod/mod_flaml.py
+++ b/Project-AutoML/automl/mod/mod_flaml.py
@@ -36,21 +36,23 @@
model_path = "flaml.pkl"
conda_env = {
- "channels": ["defaults", "conda-forge"],
+ "name": "MLflow-AutoML",
"dependencies": [
"python=3.8.2",
+ "pip",
{
"pip": [
"mlflow",
+ "click==8.0.",
"scikit-learn==0.24.2",
"boto3==1.22.2",
- "pandas==1.3.5",
+ "pandas>=1.0.0",
"setuptools<59.6.0",
- "flaml==1.0.1",
+ "auto-sklearn==0.14.6",
+ "flaml==1.0.1"
],
},
- ],
- "name": "mlflow-env",
+ ]
}
@staticmethod
diff --git a/Project-AutoML/conda.yaml b/Project-AutoML/conda.yaml
index 4691fab..56eac85 100644
--- a/Project-AutoML/conda.yaml
+++ b/Project-AutoML/conda.yaml
@@ -16,8 +16,6 @@
# under the License.
name: MLflow-AutoML
-channels:
- - https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
dependencies:
- python=3.8.2
- pip