Fix label mismatch guard, self-loop removal, lasftm typo

netfm/evaluate/pipeline.py

@@ -51,7 +51,7 @@ def run_full_evaluation(
 
     if data.y is not None:
         labels = data.y.cpu().numpy().squeeze()
-        if labels.ndim == 1 and len(np.unique(labels)) > 1:
+        if labels.ndim == 1 and len(labels) == data.num_nodes and len(np.unique(labels)) > 1:
             n = data.num_nodes
             train_mask = np.zeros(n, dtype=bool)
             test_mask = np.zeros(n, dtype=bool)
@@ -90,7 +90,7 @@ def run_full_evaluation(
 
     if data.y is not None:
         labels = data.y.cpu().numpy().squeeze()
-        if labels.ndim == 1 and len(np.unique(labels)) > 1:
+        if labels.ndim == 1 and len(labels) == data.num_nodes and len(np.unique(labels)) > 1:
             results["community_detection"] = {
                 "netfm": evaluate_community_detection(embeddings, labels),
             }