Skip to content

Commit bf52c25

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent befbb83 commit bf52c25

2 files changed

Lines changed: 5 additions & 5 deletions

File tree

examples/llm/__init__.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,7 @@
77
try:
88
from .whg_model import WarehouseGRetriever, WarehouseTaskHead # noqa: F401
99
from .whg_demo import ( # noqa: F401
10-
WarehouseConversationSystem, WHGConversationSystem
11-
)
10+
WarehouseConversationSystem, WHGConversationSystem)
1211
__all__ = [
1312
'WarehouseGRetriever',
1413
'WarehouseTaskHead',

torch_geometric/datasets/relbench.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,9 @@
6767

6868
class RelBenchProcessor:
6969
"""Utility for converting RelBench datasets to PyG HeteroData format."""
70-
def __init__(self, sbert_model: str = 'sentence-transformers/all-MiniLM-L6-v2') -> None: # noqa: E501
70+
def __init__(
71+
self, sbert_model: str = 'sentence-transformers/all-MiniLM-L6-v2'
72+
) -> None: # noqa: E501
7173
"""Initialize processor with SBERT model."""
7274
if not RELBENCH_AVAILABLE:
7375
raise ImportError('RelBench is required. Install with: '
@@ -217,8 +219,7 @@ def _add_warehouse_labels_unified(
217219
# Get table-level inference (returns single values per table)
218220
table_lineage_value = self._infer_table_lineage_value(
219221
table_name, db)
220-
table_silo_value = self._infer_table_silo_value(
221-
table_name, db)
222+
table_silo_value = self._infer_table_silo_value(table_name, db)
222223
table_anomaly_value = self._infer_table_anomaly_value(
223224
table_name, db)
224225

0 commit comments

Comments
 (0)