fix lmm

microsoft · Apr 28, 2024 · be36a57 · be36a57
1 parent b379c1e
commit be36a57
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/.github/workflows/contrib-openai.yml b/.github/workflows/contrib-openai.yml
@@ -68,7 +68,7 @@ jobs:
           AZURE_OPENAI_API_BASE: ${{ secrets.AZURE_OPENAI_API_BASE }}
           OAI_CONFIG_LIST: ${{ secrets.OAI_CONFIG_LIST }}
         run: |
-          pytest -k test/agentchat/contrib/retrievechat/ test/agentchat/contrib/retrievechat
+          pytest test/agentchat/contrib/retrievechat/ test/agentchat/contrib/retrievechat
       - name: Upload coverage to Codecov
         uses: codecov/codecov-action@v3
         with:

diff --git a/.github/workflows/contrib-tests.yml b/.github/workflows/contrib-tests.yml
@@ -295,7 +295,11 @@ jobs:
           fi
       - name: Coverage
         run: |
-          pytest test/agentchat/contrib/test_img_utils.py test/agentchat/contrib/test_lmm.py test/agentchat/contrib/test_llava.py test/agentchat/contrib/capabilities/test_image_generation_capability.py test/agentchat/contrib/capabilities/test_vision_capability.py --skip-openai
+          pytest test/agentchat/contrib/test_img_utils.py test/agentchat/contrib/test_lmm.py test/agentchat/contrib/test_llava.py test/agentchat/contrib/capabilities/test_vision_capability.py --skip-openai
+      - name: Image Gen Coverage
+        if: ${{ matrix.os != 'windows-2019' && matrix.python-version != '3.12' }}
+        run: |
+          pytest test/agentchat/contrib/capabilities/test_image_generation_capability.py --skip-openai
       - name: Upload coverage to Codecov
         uses: codecov/codecov-action@v3
         with:

diff --git a/test/agentchat/contrib/capabilities/test_image_generation_capability.py b/test/agentchat/contrib/capabilities/test_image_generation_capability.py
@@ -184,9 +184,6 @@ def test_image_generation_capability_negative(monkeypatch, image_gen_capability)
 @pytest.mark.skipif(skip_requirement, reason="Dependencies are not installed.")
 def test_image_generation_capability_cache(monkeypatch):
     """Tests ImageGeneration capability to cache the generated images."""
-    if sys.platform == "win32" and sys.version_info == (3, 12):
-        pytest.skip("Skipping test on Windows with Python 3.12")
-
     test_image_size = (256, 256)
 
     # Patching the _should_generate_image and _extract_prompt methods to avoid TextAnalyzerAgent making API calls.