Move mypy

.github/workflows/lint.yml

@@ -33,19 +33,3 @@ jobs:
         run: |
           printf "You can fix formatting by running 'black' on the modified python files:\n"
           printf "  git diff ${GITHUB_BASE_REF?} --name-only -- '*.py' ':!third_party' | xargs black\n"
-
-  mypy:
-    name: MyPy Type Checking
-    runs-on: ubuntu-latest
-    steps:
-      - name: Checking out repository
-        uses: actions/checkout@8f4b7f84864484a7bf31766abe9204da3cbe65b3 # v3.5.0
-      - name: Setting up python
-        uses: actions/setup-python@d27e3f3d7c64b4bbf8e4abfb9b63b83e846e0435 # v4.5.0
-      - name: Install MyPy
-        run: |
-          python3 -m pip install mypy==1.8.0
-          python3 -m pip install -r mypy-requirements.txt
-      - name: Type check serving/ directory
-        run: |
-          mypy serving/

.github/workflows/test.yml

@@ -45,9 +45,17 @@ jobs:
           pip install -e core[testing] -e serving[testing]
 
       - name: Run core tests
+        if: !cancelled()
         run: |
           pytest -n 4 core/
 
       - name: Run serving tests
+        if: !cancelled()
         run: |
           pytest -n 4 serving/
+
+      - name: MyPy Type Checking
+        if: !cancelled()
+        run: |
+          pip install -r mypy-requirements.txt
+          mypy serving/