pit rewrite does not need index

microsoft · Oct 26, 2023 · 8702049 · 8702049
1 parent d82ab8d
commit 8702049
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/qlib/data/storage/storage.py b/qlib/data/storage/storage.py
@@ -574,7 +574,7 @@ def write(self, data_array: np.ndarray, index: int = None):
         """
         raise NotImplementedError("Subclass of FeatureStorage must implement `write` method")
 
-    def rewrite(self, data: Union[List, np.ndarray, Tuple], index: int):
+    def rewrite(self, data: Union[List, np.ndarray, Tuple]):
         """overwrite all data in FeatureStorage with data
 
         Parameters
@@ -585,7 +585,7 @@ def rewrite(self, data: Union[List, np.ndarray, Tuple], index: int):
             data start index
         """
         self.clear()
-        self.write(data, index)
+        self.write(data, 0)
 
     def update(self, data_array: np.ndarray) -> None:
         """update data to storage, replace current data from start_date to end_date with given data_array