From 5d441563d06143a2c441d255f7fc82f143da0879 Mon Sep 17 00:00:00 2001
From: Tao Gong <gongtao950513@gmail.com>
Date: Mon, 7 Mar 2022 21:13:58 +0800
Subject: [PATCH] [Refactor]: Change _prepare_data() to prepare_data() in
BaseDataset (#108)
---
mmengine/dataset/base_dataset.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/mmengine/dataset/base_dataset.py b/mmengine/dataset/base_dataset.py
index 586292eb..0fdfbd0f 100644
--- a/mmengine/dataset/base_dataset.py
+++ b/mmengine/dataset/base_dataset.py
@@ -350,10 +350,10 @@ class BaseDataset(Dataset):
self.full_init()
if self.test_mode:
- return self._prepare_data(idx)
+ return self.prepare_data(idx)
for _ in range(self.max_refetch):
- data_sample = self._prepare_data(idx)
+ data_sample = self.prepare_data(idx)
if data_sample is None:
idx = self._rand_another()
continue
@@ -527,7 +527,7 @@ class BaseDataset(Dataset):
"""
return np.random.randint(0, len(self))
- def _prepare_data(self, idx) -> Any:
+ def prepare_data(self, idx) -> Any:
"""Get data processed by ``self.pipeline``.
Args:
--
GitLab