File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 77try :
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' ,
Original file line number Diff line number Diff line change 6767
6868class 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
You can’t perform that action at this time.
0 commit comments