Fix model coverage dep issue

PiperOrigin-RevId: 680711131
google-ai-edge · Sep 30, 2024 · d4c8646 · d4c8646
1 parent 93065dc
commit d4c8646
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/.github/workflows/model_coverage.yml b/.github/workflows/model_coverage.yml
@@ -42,15 +42,15 @@ jobs:
           python -m pip install --upgrade pip setuptools
           python -m pip cache purge
 
-      - name: Setup benchmark repository
-        run: |
-          bash ${{ github.workspace }}/benchmark/ci_ai_edge_torch/ci_setup.sh
-
       - name: Install ai-edge-torch
         run: |
           python -m pip install -r dev-requirements.txt --force-reinstall
           python -m pip install . --no-cache-dir
 
+      - name: Setup benchmark repository
+        run: |
+          bash ${{ github.workspace }}/benchmark/ci_ai_edge_torch/ci_setup.sh
+
       - name: Run tests
         run: |
           cd ${{ github.workspace }}/benchmark

diff --git a/ai_edge_torch/generative/test/test_model_conversion_large.py b/ai_edge_torch/generative/test/test_model_conversion_large.py
@@ -131,9 +131,7 @@ def test_phi2(self):
   def test_phi3(self):
     config = phi3.get_fake_model_config()
     pytorch_model = phi3.Phi3_5Mini(config).eval()
-    self._test_model(
-        config, pytorch_model, "prefill", atol=1e-5, rtol=1e-5
-    )
+    self._test_model(config, pytorch_model, "prefill", atol=1e-5, rtol=1e-5)
 
   @googletest.skipIf(
       ai_edge_config.Config.use_torch_xla,