Skip to content

Commit 90b151b

Browse files
[pre-commit.ci] pre-commit autoupdate (#352)
* [pre-commit.ci] pre-commit autoupdate updates: - [github.com/asottile/reorder-python-imports: v3.14.0 → v3.15.0](asottile/reorder-python-imports@v3.14.0...v3.15.0) * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1 parent 2ee5aba commit 90b151b

File tree

11 files changed

+21
-8
lines changed

11 files changed

+21
-8
lines changed

.pre-commit-config.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ repos:
77
- id: requirements-txt-fixer
88
- id: trailing-whitespace
99
- repo: https://github.com/asottile/reorder-python-imports
10-
rev: v3.14.0
10+
rev: v3.15.0
1111
hooks:
1212
- id: reorder-python-imports
1313
args: [--py37-plus]

examples/run_markowitz_portfolio_optimization.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,9 @@
77
from meta import config
88
from meta import config_tickers
99
from meta.data_processor import DataProcessor
10-
from meta.env_portfolio_allocation.env_portfolio_yahoofinance import StockPortfolioEnv
1110
from meta.data_processors._base import DataSource
11+
from meta.env_portfolio_allocation.env_portfolio_yahoofinance import StockPortfolioEnv
12+
1213

1314
def data_split(df, start, end, target_date_col="time"):
1415
"""

examples/run_rl_portfolio_optimization.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,9 @@
77
from meta import config
88
from meta import config_tickers
99
from meta.data_processor import DataProcessor
10-
from meta.env_portfolio_allocation.env_portfolio_yahoofinance import StockPortfolioEnv
1110
from meta.data_processors._base import DataSource
11+
from meta.env_portfolio_allocation.env_portfolio_yahoofinance import StockPortfolioEnv
12+
1213

1314
def data_split(df, start, end, target_date_col="time"):
1415
"""

meta/data_processor.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,11 @@
44

55
import numpy as np
66
import pandas as pd
7+
78
from meta.config_tickers import DOW_30_TICKER
89
from meta.data_processors._base import DataSource
910

11+
1012
class DataProcessor:
1113
def __init__(
1214
self,

meta/data_processors/_base.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
11
import os
2+
from enum import Enum
23
from typing import List
4+
35
import numpy as np
46
import pandas as pd
57
import stockstats
@@ -25,7 +27,7 @@
2527
from meta.config_tickers import SP_500_TICKER
2628
from meta.config_tickers import SSE_50_TICKER
2729
from meta.config_tickers import TECDAX_TICKER
28-
from enum import Enum
30+
2931

3032
class DataSource(Enum):
3133
akshare = "akshare"
@@ -44,6 +46,7 @@ class DataSource(Enum):
4446
wrds = "wrds"
4547
yahoofinance = "yahoofinance"
4648

49+
4750
class _Base:
4851
def __init__(
4952
self,
@@ -272,7 +275,7 @@ def add_turbulence(self):
272275
DataSource.ccxt,
273276
DataSource.iexcloud,
274277
DataSource.joinquant,
275-
DataSource.quantconnect
278+
DataSource.quantconnect,
276279
]:
277280
print(
278281
f"Turbulence not supported for {self.data_source} yet. Return original DataFrame."
@@ -282,7 +285,7 @@ def add_turbulence(self):
282285
DataSource.ricequant,
283286
DataSource.tushare,
284287
DataSource.wrds,
285-
DataSource.yahoofinance
288+
DataSource.yahoofinance,
286289
]:
287290
turbulence_index = self.calculate_turbulence()
288291
self.dataframe = self.dataframe.merge(turbulence_index, on="time")
@@ -352,7 +355,7 @@ def add_vix(self):
352355
DataSource.joinquant,
353356
DataSource.quantconnect,
354357
DataSource.ricequant,
355-
DataSource.tushare
358+
DataSource.tushare,
356359
]:
357360
print(
358361
f"VIX is not applicable for {self.data_source}. Return original DataFrame"

meta/env_future_trading/wt4elegantrl/wtpy/apps/datahelper/DHFactory.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55

66
from meta.data_processors._base import DataSource
77

8+
89
class DHFactory:
910
@staticmethod
1011
def createHelper(name: str) -> BaseDataHelper:

unit_tests/test_alphavantage.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
from meta.data_processor import DataProcessor
44
from meta.data_processors._base import DataSource
55

6+
67
def test_alphavantage():
78
TRADE_START_DATE = "2020-09-01"
89
TRADE_END_DATE = "2021-09-11"

unit_tests/test_baostock.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
from meta.data_processor import DataProcessor
44
from meta.data_processors._base import DataSource
55

6+
67
def test_baostock():
78
TRADE_START_DATE = "2020-09-01"
89
TRADE_END_DATE = "2021-09-11"
@@ -19,7 +20,7 @@ def test_baostock():
1920
]
2021
kwargs = {}
2122
p = DataProcessor(
22-
data_source= DataSource.baostock,
23+
data_source=DataSource.baostock,
2324
start_date=TRADE_START_DATE,
2425
end_date=TRADE_END_DATE,
2526
time_interval=TIME_INTERVAL,

unit_tests/test_binance.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
from meta.data_processor import DataProcessor
88
from meta.data_processor import DataSource
99

10+
1011
@pytest.fixture(scope="session")
1112
def start_date():
1213
return "2021-01-01"

unit_tests/test_quandl.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
from meta.data_processor import DataProcessor
44
from meta.data_processors._base import DataSource
55

6+
67
def test_quandl():
78
TRADE_START_DATE = "2020-09-01"
89
TRADE_END_DATE = "2021-09-11"

0 commit comments

Comments
 (0)