From 1358cef069905948c6ccfa5de59d5cc2366541e2 Mon Sep 17 00:00:00 2001
From: Tiziano <tizianocitro98@gmail.com>
Date: Tue, 28 Apr 2026 15:49:58 +0200
Subject: [PATCH] refactor: rename Node2Vec HLP module to Node2VecSLP

---
 examples/{node2vec.py => node2vecslp.py} | 14 +++++++-------
 hyperbench/hlp/__init__.py               |  4 ++--
 hyperbench/hlp/node2vec_hlp.py           |  2 +-
 3 files changed, 10 insertions(+), 10 deletions(-)
 rename examples/{node2vec.py => node2vecslp.py} (94%)

diff --git a/examples/node2vec.py b/examples/node2vecslp.py
similarity index 94%
rename from examples/node2vec.py
rename to examples/node2vecslp.py
index c0de8c8..30d608e 100644
--- a/examples/node2vec.py
+++ b/examples/node2vecslp.py
@@ -6,7 +6,7 @@
     BinaryRecall,
 )
 from hyperbench.data import AlgebraDataset, DataLoader, SamplingStrategy
-from hyperbench.hlp import Node2VecHlpModule
+from hyperbench.hlp import Node2VecSLPHlpModule
 from hyperbench.nn import Node2VecEnricher
 from hyperbench.train import MultiModelTrainer, RandomNegativeSampler
 from hyperbench.types import HData, ModelConfig
@@ -120,7 +120,7 @@
         persistent_workers=True,
     )
 
-    precomputed_node2vec_module = Node2VecHlpModule(
+    precomputed_node2vecslp_module = Node2VecSLPHlpModule(
         encoder_config={
             "mode": "precomputed",
             "num_features": num_features,
@@ -132,7 +132,7 @@
     )
 
     train_hyperedge_index = train_dataset.hdata.hyperedge_index
-    joint_node2vec_module = Node2VecHlpModule(
+    joint_node2vecslp_module = Node2VecSLPHlpModule(
         encoder_config={
             "mode": "joint",
             "num_features": num_features,
@@ -156,17 +156,17 @@
 
     configs = [
         ModelConfig(
-            name="node2vec",
+            name="node2vecslp",
             version="precomputed",
-            model=precomputed_node2vec_module,
+            model=precomputed_node2vecslp_module,
             train_dataloader=train_loader,
             val_dataloader=val_loader,
             test_dataloader=test_loader,
         ),
         ModelConfig(
-            name="node2vec",
+            name="node2vecslp",
             version="joint",
-            model=joint_node2vec_module,
+            model=joint_node2vecslp_module,
             train_dataloader=train_loader,
             val_dataloader=val_loader,
             test_dataloader=test_loader,
diff --git a/hyperbench/hlp/__init__.py b/hyperbench/hlp/__init__.py
index 9863403..7307230 100644
--- a/hyperbench/hlp/__init__.py
+++ b/hyperbench/hlp/__init__.py
@@ -5,7 +5,7 @@
 from .hlp import HlpModule
 from .hypergcn_hlp import HyperGCNHlpModule, HyperGCNEncoderConfig
 from .mlp_hlp import MLPHlpModule, MlpEncoderConfig
-from .node2vec_hlp import Node2VecEncoderConfig, Node2VecHlpModule
+from .node2vec_hlp import Node2VecEncoderConfig, Node2VecSLPHlpModule
 
 __all__ = [
     "CommonNeighborsHlpModule",
@@ -21,5 +21,5 @@
     "MlpEncoderConfig",
     "MLPHlpModule",
     "Node2VecEncoderConfig",
-    "Node2VecHlpModule",
+    "Node2VecSLPHlpModule",
 ]
diff --git a/hyperbench/hlp/node2vec_hlp.py b/hyperbench/hlp/node2vec_hlp.py
index 0e6e607..6004cbf 100644
--- a/hyperbench/hlp/node2vec_hlp.py
+++ b/hyperbench/hlp/node2vec_hlp.py
@@ -63,7 +63,7 @@ class Node2VecEncoderConfig(TypedDict):
     node2vec_loss_weight: NotRequired[float]
 
 
-class Node2VecHlpModule(HlpModule):
+class Node2VecSLPHlpModule(HlpModule):
     """
     A LightningModule for Node2Vec-based Hyperedge Link Prediction.