fix lint

Author: beinan

python/python/knowledge_graph/config.py

@@ -6,8 +6,8 @@
 from pathlib import Path
 from typing import Any, Mapping, Optional
 
-import yaml
 import pyarrow.fs
+import yaml
 from lance_graph import GraphConfig
 
 

python/python/knowledge_graph/store.py

@@ -36,7 +36,8 @@ def __init__(self, config: "KnowledgeGraphConfig"):
                 str(self._root), **(self.config.storage_options or {})
             )
         except ImportError:
-            # Re-raise explicit ImportError if protocol driver (e.g. gcsfs, s3fs) is missing
+            # Re-raise explicit ImportError if protocol driver (e.g. gcsfs, s3fs)
+            # is missing
             raise
 
     @property
@@ -55,7 +56,8 @@ def ensure_layout(self) -> None:
             self._fs.makedirs(self._fs_path, exist_ok=True)
         except Exception:
             # S3/GCS might not support directory creation or it might be implicit.
-            # We treat failure here as non-fatal if the path is actually accessible later,
+            # We treat failure here as non-fatal if the path is actually accessible
+            # later,
             # but usually makedirs is safe on object stores (no-op).
             pass
 

python/python/tests/test_gcs_integration.py

@@ -1,8 +1,9 @@
-import pyarrow as pa
-import pytest
 from pathlib import Path
 from unittest.mock import MagicMock, patch
-from knowledge_graph.config import build_graph_config_from_mapping, KnowledgeGraphConfig
+
+import pyarrow as pa
+import pytest
+from knowledge_graph.config import KnowledgeGraphConfig, build_graph_config_from_mapping
 from lance_graph import CypherQuery
 
 
@@ -89,7 +90,10 @@ def test_load_schema_payload_s3(self, mock_pyarrow):
 
         # Mock filesystem
         mock_fs = MagicMock()
-        mock_pyarrow.fs.FileSystem.from_uri.return_value = (mock_fs, "bucket/graph/graph.yaml")
+        mock_pyarrow.fs.FileSystem.from_uri.return_value = (
+            mock_fs,
+            "bucket/graph/graph.yaml",
+        )
 
         # Mock file content
         mock_input_stream = MagicMock()
@@ -100,5 +104,7 @@ def test_load_schema_payload_s3(self, mock_pyarrow):
         payload = config._load_schema_payload()
 
         assert payload == {"nodes": {"Person": "id"}}
-        mock_pyarrow.fs.FileSystem.from_uri.assert_called_with("s3://bucket/graph/graph.yaml")
+        mock_pyarrow.fs.FileSystem.from_uri.assert_called_with(
+            "s3://bucket/graph/graph.yaml"
+        )
         mock_fs.open_input_stream.assert_called_with("bucket/graph/graph.yaml")

python/python/tests/test_knowledge_graph_config.py

@@ -360,8 +360,7 @@ class TestS3Support:
     @pytest.fixture
     def s3_config(self):
         return KnowledgeGraphConfig(
-            storage_path="s3://bucket/graph",
-            schema_path="s3://bucket/graph/graph.yaml"
+            storage_path="s3://bucket/graph", schema_path="s3://bucket/graph/graph.yaml"
         )
 
     @pytest.fixture