From 0187fc7891fd09f7a45bdba2c338b6883571fc06 Mon Sep 17 00:00:00 2001
From: Antonios Sarikas <antonios.sarikas@gmail.com>
Date: Sat, 4 Jan 2025 19:48:22 +0200
Subject: [PATCH] docs(PCDDataset): add information for shape of `y`

---
 src/aidsorb/data.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/aidsorb/data.py b/src/aidsorb/data.py
index 9d1f550..38d81c3 100644
--- a/src/aidsorb/data.py
+++ b/src/aidsorb/data.py
@@ -333,7 +333,7 @@ class PCDDataset(Dataset):
 
     .. note::
         * ``x`` and ``y`` are tensors of ``dtype=torch.float``.
-        * ``y`` has shape ``(len(labels),)``.
+        * ``y`` has shape ``(len(labels),)`` if ``transform_y=None``.
         * ``transform_x`` and ``transform_y`` expect :class:`~torch.Tensor` as
           input.