Skip to content

Commit ef11a9d

Browse files
committedMay 28, 2021
modify the default value of exists_skip in the GetData.qlib_data parameter to False
1 parent 98eacf8 commit ef11a9d

File tree

10 files changed

+17
-10
lines changed

10 files changed

+17
-10
lines changed
 

‎examples/highfreq/workflow.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ def _init_qlib(self):
8585
# use yahoo_cn_1min data
8686
QLIB_INIT_CONFIG = {**HIGH_FREQ_CONFIG, **self.SPEC_CONF}
8787
provider_uri = QLIB_INIT_CONFIG.get("provider_uri")
88-
GetData().qlib_data(target_dir=provider_uri, interval="1min", region=REG_CN)
88+
GetData().qlib_data(target_dir=provider_uri, interval="1min", region=REG_CN, exists_skip=True)
8989
qlib.init(**QLIB_INIT_CONFIG)
9090

9191
def _prepare_calender_cache(self):

‎examples/hyperparameter/LightGBM/hyperparameter_158.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ def objective(trial):
3737
if __name__ == "__main__":
3838

3939
provider_uri = "~/.qlib/qlib_data/cn_data"
40-
GetData().qlib_data(target_dir=provider_uri, region=REG_CN)
40+
GetData().qlib_data(target_dir=provider_uri, region=REG_CN, exists_skip=True)
4141
qlib.init(provider_uri=provider_uri, region="cn")
4242

4343
dataset = init_instance_by_config(CSI300_DATASET_CONFIG)

‎examples/hyperparameter/LightGBM/hyperparameter_360.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ def objective(trial):
4040
if __name__ == "__main__":
4141

4242
provider_uri = "~/.qlib/qlib_data/cn_data"
43-
GetData().qlib_data(target_dir=provider_uri, region=REG_CN)
43+
GetData().qlib_data(target_dir=provider_uri, region=REG_CN, exists_skip=True)
4444
qlib.init(provider_uri=provider_uri, region=REG_CN)
4545

4646
dataset = init_instance_by_config(DATASET_CONFIG)

‎examples/model_interpreter/feature.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414

1515
# use default data
1616
provider_uri = "~/.qlib/qlib_data/cn_data" # target_dir
17-
GetData().qlib_data(target_dir=provider_uri, region=REG_CN)
17+
GetData().qlib_data(target_dir=provider_uri, region=REG_CN, exists_skip=True)
1818

1919
qlib.init(provider_uri=provider_uri, region=REG_CN)
2020

‎examples/rolling_process_data/workflow.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ def _init_qlib(self):
2323
"""initialize qlib"""
2424
# use yahoo_cn_1min data
2525
provider_uri = "~/.qlib/qlib_data/cn_data" # target_dir
26-
GetData().qlib_data(target_dir=provider_uri, region=REG_CN)
26+
GetData().qlib_data(target_dir=provider_uri, region=REG_CN, exists_skip=True)
2727
qlib.init(provider_uri=provider_uri, region=REG_CN)
2828

2929
def _dump_pre_handler(self, path):

‎examples/run_all_model.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@
3737
},
3838
}
3939

40-
GetData().qlib_data(target_dir=provider_uri, region=REG_CN)
40+
GetData().qlib_data(target_dir=provider_uri, region=REG_CN, exists_skip=True)
4141
qlib.init(provider_uri=provider_uri, region=REG_CN, exp_manager=exp_manager)
4242

4343
# decorator to check the arguments

‎examples/workflow_by_code.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414

1515
# use default data
1616
provider_uri = "~/.qlib/qlib_data/cn_data" # target_dir
17-
GetData().qlib_data(target_dir=provider_uri, region=REG_CN)
17+
GetData().qlib_data(target_dir=provider_uri, region=REG_CN, exists_skip=True)
1818
qlib.init(provider_uri=provider_uri, region=REG_CN)
1919

2020
port_analysis_config = {

‎qlib/tests/__init__.py

+1
Original file line numberDiff line numberDiff line change
@@ -19,5 +19,6 @@ def setUpClass(cls) -> None:
1919
interval="1d",
2020
target_dir=cls.provider_uri,
2121
delete_old=False,
22+
exists_skip=True,
2223
)
2324
init(provider_uri=cls.provider_uri, region=REG_CN, **cls._setup_kwargs)

‎qlib/tests/data.py

+6-2
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,7 @@ def qlib_data(
113113
interval="1d",
114114
region="cn",
115115
delete_old=True,
116-
exists_skip=True,
116+
exists_skip=False,
117117
):
118118
"""download cn qlib data from remote
119119
@@ -132,7 +132,7 @@ def qlib_data(
132132
delete_old: bool
133133
delete an existing directory, by default True
134134
exists_skip: bool
135-
exists skip, by default True
135+
exists skip, by default False
136136
137137
Examples
138138
---------
@@ -145,6 +145,10 @@ def qlib_data(
145145
146146
"""
147147
if exists_skip and exists_qlib_data(target_dir):
148+
logger.warning(
149+
f"Data already exists: {target_dir}, the data download will be skipped\n"
150+
f"\tIf downloading is required: `exists_skip=False` or `change target_dir`"
151+
)
148152
return
149153

150154
qlib_version = ".".join(re.findall(r"(\d+)\.+", qlib.__version__))

‎tests/test_get_data.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,9 @@ def tearDownClass(cls) -> None:
3434

3535
def test_0_qlib_data(self):
3636

37-
GetData().qlib_data(name="qlib_data_simple", target_dir=QLIB_DIR, region="cn", interval="1d", delete_old=False)
37+
GetData().qlib_data(
38+
name="qlib_data_simple", target_dir=QLIB_DIR, region="cn", interval="1d", delete_old=False, exists_skip=True
39+
)
3840
df = D.features(D.instruments("csi300"), self.FIELDS)
3941
self.assertListEqual(list(df.columns), self.FIELDS, "get qlib data failed")
4042
self.assertFalse(df.dropna().empty, "get qlib data failed")

0 commit comments

Comments
 (0)
Please sign in to comment.