style: satisfy pre-commit gate

Author: teng-lin

scripts/check_rpc_health.py

@@ -148,7 +148,7 @@ def extract_id_recursive(data: Any) -> str | None:
     """
     if data is None:
         return None
-    if isinstance(data, (str, int)):
+    if isinstance(data, str | int):
         return str(data)
     if isinstance(data, list) and len(data) > 0:
         return extract_id_recursive(data[0])

src/notebooklm/_core.py

@@ -60,7 +60,10 @@ def is_auth_error(error: Exception) -> bool:
 
     # Don't treat network/rate limit/server errors as auth errors
     # even if they're subclasses of RPCError
-    if isinstance(error, (NetworkError, RPCTimeoutError, RateLimitError, ServerError, ClientError)):
+    if isinstance(
+        error,
+        NetworkError | RPCTimeoutError | RateLimitError | ServerError | ClientError,
+    ):
         return False
 
     # HTTP 401/403 are auth errors

tests/e2e/conftest.py

@@ -62,9 +62,10 @@ def assert_generation_started(result, artifact_type: str = "Artifact") -> None:
         pytest.skip("Rate limited by API")
 
     assert result.task_id, f"{artifact_type} generation failed: {result.error}"
-    assert result.status in ("pending", "in_progress"), (
-        f"Unexpected {artifact_type.lower()} status: {result.status}"
-    )
+    assert result.status in (
+        "pending",
+        "in_progress",
+    ), f"Unexpected {artifact_type.lower()} status: {result.status}"
 
 
 def has_auth() -> bool:

tests/e2e/test_chat.py

@@ -307,9 +307,9 @@ async def test_reference_source_ids_exist_in_notebook(self, client, multi_source
 
         # All reference source IDs should exist in the notebook
         for ref in result.references:
-            assert ref.source_id in source_ids, (
-                f"Reference source_id {ref.source_id} not found in notebook sources"
-            )
+            assert (
+                ref.source_id in source_ids
+            ), f"Reference source_id {ref.source_id} not found in notebook sources"
 
     @pytest.mark.asyncio
     async def test_cited_text_matches_source_content(self, client, multi_source_notebook_id):

tests/e2e/test_research_import_verification.py

@@ -127,8 +127,8 @@ def test_import_sources_filters_empty_urls():
         valid_sources = [s for s in sources_mixed if s.get("url")]
 
         # Should only have 1 valid source
-        assert len(valid_sources) == 1, (
-            f"Expected 1 valid source after filtering, got {len(valid_sources)}"
-        )
+        assert (
+            len(valid_sources) == 1
+        ), f"Expected 1 valid source after filtering, got {len(valid_sources)}"
         assert valid_sources[0]["url"] == "https://example.com/valid"
         assert valid_sources[0]["title"] == "Valid Source"

tests/e2e/test_source_selection.py

@@ -215,9 +215,9 @@ async def test_notebook_has_multiple_sources(self, client, multi_source_notebook
         """Verify the test notebook has at least 3 sources."""
         sources = await client.sources.list(multi_source_notebook_id)
 
-        assert len(sources) >= 3, (
-            f"Expected at least 3 sources for multi-source tests, got {len(sources)}"
-        )
+        assert (
+            len(sources) >= 3
+        ), f"Expected at least 3 sources for multi-source tests, got {len(sources)}"
 
         # Verify each source has an ID
         for source in sources:

tests/integration/cli_vcr/test_artifacts.py

@@ -29,7 +29,7 @@ def test_artifact_list(self, runner, mock_auth_for_vcr, mock_context, json_flag)
         if json_flag and result.exit_code == 0:
             data = parse_json_output(result.output)
             assert data is not None, "Expected valid JSON output"
-            assert isinstance(data, (list, dict))
+            assert isinstance(data, list | dict)
 
 
 class TestArtifactListByType:

tests/integration/cli_vcr/test_chat.py

@@ -32,7 +32,7 @@ def test_ask_question_json(self, runner, mock_auth_for_vcr, mock_context):
         if result.exit_code == 0:
             data = parse_json_output(result.output)
             assert data is not None, "Expected valid JSON output"
-            assert isinstance(data, (list, dict))
+            assert isinstance(data, list | dict)
 
 
 class TestHistoryCommand:

tests/integration/cli_vcr/test_notebooks.py

@@ -32,7 +32,7 @@ def test_list_notebooks_json(self, runner, mock_auth_for_vcr):
 
         data = parse_json_output(result.output)
         assert data is not None, "Expected valid JSON output"
-        assert isinstance(data, (list, dict))
+        assert isinstance(data, list | dict)
 
 
 class TestSummaryCommand:

tests/integration/cli_vcr/test_sources.py

@@ -29,7 +29,7 @@ def test_source_list(self, runner, mock_auth_for_vcr, mock_context, json_flag):
         if json_flag and result.exit_code == 0:
             data = parse_json_output(result.output)
             assert data is not None, "Expected valid JSON output"
-            assert isinstance(data, (list, dict))
+            assert isinstance(data, list | dict)
 
 
 class TestSourceAddCommand: