add csi800 and csiall

chenditc · Dec 13, 2022 · f41a11c · f41a11c
1 parent a906e4c
commit f41a11c
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/qlib/dump_index_weight.py b/qlib/dump_index_weight.py
@@ -11,7 +11,9 @@ def dump_all_to_sqlib_source(skip_exists=False):
 
   index_map = {
     "csi300" : "399300.SZ",
-    "csi500" : "000905.SH"
+    "csi500" : "000905.SH",
+    "csi800" : "000906.SH",
+    "csiall" : "000985.SH",
   }
 
   script_path = os.path.dirname(os.path.realpath(__file__))
@@ -49,4 +51,4 @@ def dump_all_to_sqlib_source(skip_exists=False):
     pd.concat(result_df_list).to_csv(filename, index=False, header=False, sep='\t')
 
 if __name__ == "__main__":
-  fire.Fire(dump_all_to_sqlib_source)
+  fire.Fire(dump_all_to_sqlib_source)