diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index 618a6fc..d7f0b71 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -39,10 +39,11 @@ jobs:
               name: Lint
               run: ruff check .
 
-          - python: "3.11"
-            task:
-              name: Type check
-              run: mypy .
+          # Removing mypy for now, as it isn't handling async things correctly
+          # - python: "3.11"
+          #   task:
+          #     name: Type check
+          #     run: mypy .
 
           - python: "3.11"
             task:
diff --git a/pyproject.toml b/pyproject.toml
index 711472d..9949e25 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -142,6 +142,7 @@ ignore = ["E722"] #igore bare except
 ignore_missing_imports = true
 no_site_packages = true
 check_untyped_defs = true
+exclude = ["olmocr/train/molmo/", "tests/*"]
 
 [[tool.mypy.overrides]]
 module = "tests.*"