From bb224cd3dd48b3a987dfde9fc5b56d817a8f4eea Mon Sep 17 00:00:00 2001
From: Natalia Razinkov <97098247+NataliRa@users.noreply.github.com>
Date: Sun, 26 Jun 2022 15:15:51 +0300
Subject: [PATCH] Bug fix for PytorchData dataset

---
 apt/utils/datasets/datasets.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/apt/utils/datasets/datasets.py b/apt/utils/datasets/datasets.py
index 6569b37..a386226 100644
--- a/apt/utils/datasets/datasets.py
+++ b/apt/utils/datasets/datasets.py
@@ -292,7 +292,7 @@ class PytorchData(Dataset):
         :type idx: int
         :return: the sample as a pytorch Tensor
         """
-        return self.x[idx]
+        return self._x[idx]
 
     def get_item(self, idx: int) -> Tensor:
         """
@@ -302,11 +302,11 @@ class PytorchData(Dataset):
         :type idx: int
         :return: the sample and label as pytorch Tensors. Returned as a tuple (sample, label)
         """
-        sample, label = self.x[idx], self.y[idx]
+        sample, label = self._x[idx], self._y[idx]
         return sample, label
 
     def __len__(self):
-        return len(self.x)
+        return len(self._x)
 
 
 class DatasetFactory: