From f17d1d02798afedd01825c38a9ad12a2922d1594 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Niklas=20B=C3=B6hm?= <jan-niklas.boehm@uni-tuebingen.de>
Date: Tue, 2 Jan 2024 17:51:19 +0100
Subject: [PATCH] Add comment

---
 tsimcne/tsimcne.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tsimcne/tsimcne.py b/tsimcne/tsimcne.py
index ee76935..1047042 100644
--- a/tsimcne/tsimcne.py
+++ b/tsimcne/tsimcne.py
@@ -559,6 +559,8 @@ def transform(
         elif not return_labels and return_backbone_feat:
             return Y, backbone_features
         elif return_labels and not return_backbone_feat:
+            # XXX: this for some reason changes the labels; but I
+            # don't know what causes this!
             labels = torch.hstack([lbl for _, lbl in loader])
             return Y, labels
         else: