Fix RAI image builds

Azure · Nov 27, 2024 · e8bdfb1 · e8bdfb1
1 parent 57da35d
commit e8bdfb1
Show file tree

Hide file tree

Showing 8 changed files with 50 additions and 8 deletions.
diff --git a/assets/responsibleai/environments/responsibleai-tabular/asset.yaml b/assets/responsibleai/environments/responsibleai-tabular/asset.yaml
@@ -6,6 +6,6 @@ extra_config: environment.yaml
 test:
   pytest:
     enabled: true 
-    pip_requirements: tests/requirements.txt
+    conda_environment: tests/test_conda_env.yaml
     tests_dir: tests
 categories: ["Responsible AI"]
diff --git a/assets/responsibleai/environments/responsibleai-tabular/tests/requirements.txt b/assets/responsibleai/environments/responsibleai-tabular/tests/requirements.txt
@@ -1,4 +1,4 @@
-azure-ai-ml~=1.0.0
-azure.identity==1.10.0
-azureml-core~=1.53.0
+azure-ai-ml==1.22.4
+azure.identity==1.19.0
+azureml-core>=1.58.0
 azureml-mlflow
diff --git a/assets/responsibleai/environments/responsibleai-tabular/tests/test_conda_env.yaml b/assets/responsibleai/environments/responsibleai-tabular/tests/test_conda_env.yaml
@@ -0,0 +1,14 @@
+channels:
+  - conda-forge
+  - defaults
+  - anaconda
+dependencies:
+  - python=3.12.0
+  - pip
+  - pip:
+    - -r requirements.txt
+    ## Test requirements
+    - pytest
+    - tqdm
+    - azure-core
+name: rai_test
diff --git a/assets/responsibleai/environments/responsibleai-text/asset.yaml b/assets/responsibleai/environments/responsibleai-text/asset.yaml
@@ -5,7 +5,7 @@ spec: spec.yaml
 extra_config: environment.yaml
 test:
   pytest:
-    enabled: true 
-    pip_requirements: tests/requirements.txt
+    enabled: true
+    conda_environment: tests/test_conda_env.yaml
     tests_dir: tests
 categories: ["Responsible AI"]
diff --git a/assets/responsibleai/environments/responsibleai-text/tests/test_conda_env.yaml b/assets/responsibleai/environments/responsibleai-text/tests/test_conda_env.yaml
@@ -0,0 +1,14 @@
+channels:
+  - conda-forge
+  - defaults
+  - anaconda
+dependencies:
+  - python=3.12.0
+  - pip
+  - pip:
+    - -r requirements.txt
+    ## Test requirements
+    - pytest
+    - tqdm
+    - azure-core
+name: rai_test
diff --git a/assets/responsibleai/environments/responsibleai-vision/asset.yaml b/assets/responsibleai/environments/responsibleai-vision/asset.yaml
@@ -6,6 +6,6 @@ extra_config: environment.yaml
 test:
   pytest:
     enabled: true 
-    pip_requirements: tests/requirements.txt
+    conda_environment: tests/test_conda_env.yaml
     tests_dir: tests
 categories: ["Responsible AI"]
diff --git a/assets/responsibleai/environments/responsibleai-vision/tests/test_conda_env.yaml b/assets/responsibleai/environments/responsibleai-vision/tests/test_conda_env.yaml
@@ -0,0 +1,14 @@
+channels:
+  - conda-forge
+  - defaults
+  - anaconda
+dependencies:
+  - python=3.12.0
+  - pip
+  - pip:
+    - -r requirements.txt
+    ## Test requirements
+    - pytest
+    - tqdm
+    - azure-core
+name: rai_test
diff --git a/assets/training/automl/environments/ai-ml-automl/tests/automl_sample_test.py b/assets/training/automl/environments/ai-ml-automl/tests/automl_sample_test.py
@@ -26,7 +26,7 @@ def test_azure_ai_ml_automl():
     subscription_id = os.environ.get("subscription_id")
     resource_group = os.environ.get("resource_group")
     workspace_name = os.environ.get("workspace")
-
+    # test
     ml_client = MLClient(
         AzureCliCredential(), subscription_id, resource_group, workspace_name
     )