Fix pyright inclusions/exclusions to make the CI green

It seems the working-directory parameter in the CI config is ignored if
pyproject.toml configures pyright, so tweak that instead.

Author: smarter

.github/workflows/build.yml

@@ -21,17 +21,10 @@ jobs:
           pip install -e ".[dev,faiss]"
       - name: Run EKFAC tests that work without GPU
         run: pytest -sv tests/ekfac_tests
-      # TODO: run pyright on whole codebase
-      - name: Type Checking bergson/hessians
+      - name: Type Checking
         uses: jakebailey/pyright-action@v1
         with:
           version: 1.1.406
-          working-directory: bergson/hessians
-      - name: Type Checking tests/ekfac_tests
-        uses: jakebailey/pyright-action@v1
-        with:
-          version: 1.1.406
-          working-directory: tests/ekfac_tests
       - name: build
         run: pip wheel --no-deps -w dist .
 env:

pyproject.toml

@@ -40,7 +40,14 @@ faiss = [
 bergson = "bergson.__main__:main"
 
 [tool.pyright]
-include = ["bergson*"]
+include = ["bergson*", "tests/ekfac_tests"]
+## Temporary excludes before we merge ekfac into the main branch
+exclude = [
+    "bergson/attributor.py",
+    "bergson/collection.py",
+    "bergson/gradients.py",
+    "bergson/utils.py",
+]
 reportPrivateImportUsage = false
 
 [tool.setuptools.packages.find]