From 5b7b48e376e49f2fbfaf43f041d94b247ef9fe6e Mon Sep 17 00:00:00 2001
From: Dong Zhou <Zhou.Dong@microsoft.com>
Date: Thu, 22 Jul 2021 20:16:31 +0800
Subject: [PATCH] clean up

---
 qlib/contrib/data/dataset.py | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/qlib/contrib/data/dataset.py b/qlib/contrib/data/dataset.py
index cfda942f4..af4893acf 100644
--- a/qlib/contrib/data/dataset.py
+++ b/qlib/contrib/data/dataset.py
@@ -225,8 +225,6 @@ class MTSDatasetH(DatasetH):
             raise ValueError("cannot assign data as `num_states==0`")
         if isinstance(vals, torch.Tensor):
             vals = vals.detach().cpu().numpy()
-        # if isinstance(index, pd.Series):
-        #     index = index.index  # daily batch use Series to store index
         self._memory[index] = vals
 
     def clear_memory(self):
@@ -234,7 +232,6 @@ class MTSDatasetH(DatasetH):
             raise ValueError("cannot clear memory as `num_states==0`")
         self._memory[:] = 0
 
-    # TODO: better train/eval mode design
     def train(self):
         """enable traning mode"""
         self.batch_size, self.n_samples, self.drop_last, self.shuffle = self.params