[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

examples/llm/relbench_warehouse_demo.py

@@ -16,16 +16,17 @@
 """
 import argparse
 import time
-from typing import Dict, Any, Tuple
+from typing import Any, Dict, Tuple
 
 import torch
+
 from torch_geometric import seed_everything
 from torch_geometric.data import HeteroData
 from torch_geometric.datasets.relbench import (
     RelBenchProcessor,
     create_relbench_hetero_data,
     get_warehouse_task_info,
-    prepare_for_gretriever
+    prepare_for_gretriever,
 )
 
 
@@ -44,11 +45,9 @@ def demonstrate_basic_usage(dataset_name: str,
     start_time = time.time()
 
     # Create HeteroData with warehouse labels
-    hetero_data = create_relbench_hetero_data(
-        dataset_name=dataset_name,
-        sample_size=sample_size,
-        add_warehouse_labels=True
-    )
+    hetero_data = create_relbench_hetero_data(dataset_name=dataset_name,
+                                              sample_size=sample_size,
+                                              add_warehouse_labels=True)
 
     conversion_time = time.time() - start_time
     print(f"✅ Conversion completed in {conversion_time:.2f}s")
@@ -210,9 +209,10 @@ def main(dataset_name: str, sample_size: int, sbert_model: str,
                         help='RelBench dataset name (default: rel-trial)')
     parser.add_argument('--sample_size', type=int, default=100,
                         help='Number of records to sample (default: 100)')
-    parser.add_argument('--sbert_model', type=str, default='all-MiniLM-L6-v2',
-                        help='SBERT model for embeddings '
-                             '(default: all-MiniLM-L6-v2)')
+    parser.add_argument(
+        '--sbert_model', type=str, default='all-MiniLM-L6-v2',
+        help='SBERT model for embeddings '
+        '(default: all-MiniLM-L6-v2)')
     parser.add_argument('--save_results', action='store_true',
                         help='Save demonstration results to file')
     parser.add_argument('--seed', type=int, default=42,
@@ -221,7 +221,7 @@ def main(dataset_name: str, sample_size: int, sbert_model: str,
     args = parser.parse_args()
 
     start_time = time.time()
-    main(args.dataset, args.sample_size, args.sbert_model,
-         args.save_results, args.seed)
+    main(args.dataset, args.sample_size, args.sbert_model, args.save_results,
+         args.seed)
     total_time = time.time() - start_time
     print(f"\nTotal execution time: {total_time:.2f}s")