Merge pull request #95 from harishmohanraj/mypy-check

Fix pre-commit issues
harishmohanraj · Jan 22, 2025 · 4e807ec · 4e807ec
2 parents cda251b + f32e0a6
commit 4e807ec
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -284,6 +284,7 @@ exclude = [
     # This file needs to be either upgraded or removed and therefore should be
     # ignore from type checking for now
     "math_utils\\.py$",
+    "setup_*.py",
 ]
 
 [tool.ruff.lint]

diff --git a/test/website/test_process_notebooks.py b/test/website/test_process_notebooks.py
@@ -7,7 +7,7 @@
 import tempfile
 import textwrap
 from pathlib import Path
-from typing import Generator, Union
+from typing import Generator, Optional, Union
 
 import pytest
 
@@ -70,7 +70,7 @@ def test_cleanup_tmp_dirs_if_no_metadata() -> None:
 
 class TestAddFrontMatterToMetadataMdx:
     def test_without_metadata_mdx(self) -> None:
-        front_matter_dict: dict[str, Union[str, Union[list[str] | None]]] = {
+        front_matter_dict: dict[str, Union[str, Optional[Union[list[str]]]]] = {
             "title": "some title",
             "link": "/notebooks/some-title",
             "description": "some description",
@@ -120,7 +120,7 @@ def test_without_metadata_mdx(self) -> None:
             )
 
     def test_with_metadata_mdx(self) -> None:
-        front_matter_dict: dict[str, Union[str, Union[list[str] | None]]] = {
+        front_matter_dict: dict[str, Optional[Union[str, Union[list[str]]]]] = {
             "title": "some title",
             "link": "/notebooks/some-title",
             "description": "some description",