rqalpha 5.4.3__tar.gz → 5.5.0__tar.gz

{rqalpha-5.4.3/rqalpha.egg-info → rqalpha-5.5.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rqalpha
-Version: 5.4.3
+Version: 5.5.0
 Summary: Ricequant Algorithm Trading System
 Home-page: https://github.com/ricequant/rqalpha
 Author: ricequant

{rqalpha-5.4.3 → rqalpha-5.5.0}/rqalpha/_version.py

@@ -8,11 +8,11 @@ import json
 
 version_json = '''
 {
- "date": "2024-11-29T17:58:29+0800",
+ "date": "2025-01-08T15:55:09+0800",
  "dirty": false,
  "error": null,
- "full-revisionid": "29a200367f13ec777ae5f96bafcd6da3add9007d",
- "version": "5.4.3"
+ "full-revisionid": "ebf151930b3fa3d09f3733f706d8e44f2e169bcd",
+ "version": "5.5.0"
 }
 '''  # END VERSION_JSON
 

{rqalpha-5.4.3 → rqalpha-5.5.0}/rqalpha/const.py

@@ -201,6 +201,7 @@ class EXCHANGE(CustomEnum):
     CZCE = "CZCE"
     CFFEX = "CFFEX"
     SGEX = "SGEX"
+    BJSE = "BJSE"
 
 
 # noinspection PyPep8Naming

{rqalpha-5.4.3 → rqalpha-5.5.0}/rqalpha/data/base_data_source/data_source.py

@@ -100,8 +100,6 @@ class BaseDataSource(AbstractDataSource):
                 instruments.append(Instrument(
                     i, 
                     lambda i: self._future_info_store.get_tick_size(i),
-                    # lambda i, dt: env.data_proxy.get_futures_trading_parameters(i, dt).long_margin_ratio,
-                    # lambda i, dt: env.data_proxy.get_futures_trading_parameters(i, dt).short_margin_ratio
                     ))
         for ins_type in self.DEFAULT_INS_TYPES:
             self.register_instruments_store(InstrumentStore(instruments, ins_type))

{rqalpha-5.4.3 → rqalpha-5.5.0}/rqalpha/data/bundle.py

@@ -419,11 +419,12 @@ class UpdateDayBarTask(DayBarTask):
                     h5.close()
 
 
-def process_init(args: Optional[Synchronized] = None):
+def process_init(args: Optional[Synchronized] = None, kwargs = None):
+    kwargs = kwargs or {}
     import warnings
     with warnings.catch_warnings(record=True):
         # catch warning: rqdatac is already inited. Settings will be changed
-        rqdatac.init()
+        rqdatac.init(**kwargs)
     init_logger()
     # Initialize process shared variables
     if args:
@@ -431,17 +432,13 @@ def process_init(args: Optional[Synchronized] = None):
         sval = args
 
 
-def update_bundle(path, create, enable_compression=False, concurrency=1):
+def update_bundle(path, create, enable_compression=False, concurrency=1, **kwargs):
     if create:
         _DayBarTask = GenerateDayBarTask
     else:
         _DayBarTask = UpdateDayBarTask
 
     init_logger()
-    kwargs = {}
-    if enable_compression:
-        kwargs['compression'] = 9
-
     day_bar_args = (
         ("stocks.h5", rqdatac.all_instruments('CS').order_book_id.tolist(), STOCK_FIELDS),
         ("indexes.h5", rqdatac.all_instruments('INDX').order_book_id.tolist(), INDEX_FIELDS),
@@ -458,8 +455,11 @@ def update_bundle(path, create, enable_compression=False, concurrency=1):
 
     succeed = multiprocessing.Value(c_bool, True)
     with ProgressedProcessPoolExecutor(
-            max_workers=concurrency, initializer=process_init, initargs=(succeed, )
+            max_workers=concurrency, initializer=process_init, initargs=(succeed, kwargs)
     ) as executor:
+        kwargs = {}
+        if enable_compression:
+            kwargs['compression'] = 9
         # windows上子进程需要执行rqdatac.init, 其他os则需要执行rqdatac.reset; rqdatac.init包含了rqdatac.reset的功能
         for func in gen_file_funcs:
             executor.submit(GenerateFileTask(func), path)

{rqalpha-5.4.3 → rqalpha-5.5.0}/rqalpha/mod/rqalpha_mod_sys_analyser/mod.py

@@ -116,21 +116,9 @@ class AnalyserMod(AbstractMod):
 
             self._plot_store = PlotStore(env)
             export_as_api(self._plot_store.plot)
-
-    def get_benchmark_daily_returns(self):
-        if self._benchmark is None:
-            return np.nan
-        daily_return_list = []
-        weights = 0
-        for benchmark in self._benchmark:
-            bar = self._env.data_proxy.get_bar(benchmark[0], self._env.calendar_dt, '1d')
-            if bar.close != bar.close:
-                daily_return_list.append((0.0, benchmark[1]))
-            else:
-                daily_return_list.append((bar.close / bar.prev_close - 1.0, benchmark[1]))
-            weights += benchmark[1]
-        return sum([daily[0] * daily[1] / weights for daily in daily_return_list])
     
+    NULL_OID = {"null", "NULL"}
+
     def generate_benchmark_daily_returns_and_portfolio(self, event):
         _s = self._env.config.base.start_date
         _e = self._env.config.base.end_date
@@ -143,37 +131,41 @@ class AnalyserMod(AbstractMod):
         self._benchmark_daily_returns = np.zeros(len(trading_dates))
         weights = 0
         for order_book_id, weight in self._benchmark:
-            ins = self._env.data_proxy.instrument(order_book_id)
-            if ins is None:
-                raise RuntimeError(
-                    _("benchmark {} not exists, please entry correct order_book_id").format(order_book_id)
-                )
-            bars = self._env.data_proxy.history_bars(
-                order_book_id = order_book_id,
-                bar_count = len(trading_dates) + 1,  # Get an extra day for calculation
-                frequency = "1d",
-                field = ["datetime", "close"],
-                dt = _e,
-                skip_suspended=False,
-            )
-            if len(bars) == len(trading_dates) + 1:
-                if convert_int_to_date(bars[1]['datetime']).date() != _s:
-                    raise RuntimeError(_(
-                        "benchmark {} missing data between backtest start date {} and end date {}").format(order_book_id, _s, _e)
-                    )
-                daily_returns = (bars['close'] / np.roll(bars['close'], 1) - 1.0)[1: ]
-                self._benchmark_daily_returns = self._benchmark_daily_returns + daily_returns * weight
-                weights += weight
+            if order_book_id in self.NULL_OID:
+                daily_returns = np.zeros(len(trading_dates))
             else:
-                if len(bars) == 0:
-                    (available_s, available_e) = (ins.listed_date, ins.de_listed_date)
-                else:
-                    (available_s, available_e) = (convert_int_to_date(bars[0]['datetime']).date(), convert_int_to_date(bars[-1]['datetime']).date())
-                raise RuntimeError(
-                    _("benchmark {} available data start date {} >= backtest start date {} or end date {} <= backtest end "
-                    "date {}").format(order_book_id, available_s, _s, available_e, _e)
+                ins = self._env.data_proxy.instrument(order_book_id)
+                if ins is None:
+                    raise RuntimeError(
+                        _("benchmark {} not exists, please entry correct order_book_id").format(order_book_id)
+                    )
+                bars = self._env.data_proxy.history_bars(
+                    order_book_id = order_book_id,
+                    bar_count = len(trading_dates) + 1,  # Get an extra day for calculation
+                    frequency = "1d",
+                    field = ["datetime", "close"],
+                    dt = _e,
+                    skip_suspended=False,
                 )
-        self._benchmark_daily_returns = self._benchmark_daily_returns / weight
+                if len(bars) == len(trading_dates) + 1:
+                    if convert_int_to_date(bars[1]['datetime']).date() != _s:
+                        raise RuntimeError(_(
+                            "benchmark {} missing data between backtest start date {} and end date {}").format(order_book_id, _s, _e)
+                        )
+                    daily_returns = (bars['close'] / np.roll(bars['close'], 1) - 1.0)[1: ]
+                else:
+                    if len(bars) == 0:
+                        (available_s, available_e) = (ins.listed_date, ins.de_listed_date)
+                    else:
+                        (available_s, available_e) = (convert_int_to_date(bars[0]['datetime']).date(), convert_int_to_date(bars[-1]['datetime']).date())
+                    raise RuntimeError(
+                        _("benchmark {} available data start date {} >= backtest start date {} or end date {} <= backtest end "
+                        "date {}").format(order_book_id, available_s, _s, available_e, _e)
+                    )
+            self._benchmark_daily_returns = self._benchmark_daily_returns + daily_returns * weight
+            weights += weight
+
+        self._benchmark_daily_returns = self._benchmark_daily_returns / weights
         
         # generate benchmark portfolio
         unit_net_value = (self._benchmark_daily_returns + 1).cumprod()
@@ -229,7 +221,8 @@ class AnalyserMod(AbstractMod):
         benchmark_list = benchmarks.split(',')
         if len(benchmark_list) == 1:
             if len(benchmark_list[0].split(':')) > 1:
-                result.append((benchmark_list[0].split(':')[0], 1.0))
+                oid, weight = benchmark_list[0].split(':')
+                result.append((oid, float(weight)))
                 return result
             result.append((benchmark_list[0], 1.0))
             return result
@@ -360,7 +353,7 @@ class AnalyserMod(AbstractMod):
                 summary["benchmark_symbol"] = self._env.data_proxy.instrument(benchmark_obid).symbol
             else:
                 summary["benchmark"] = ",".join(f"{o}:{w}" for o, w in self._benchmark)
-                summary["benchmark_symbol"] = ",".join(f"{self._env.data_proxy.instrument(o).symbol}:{w}" for o, w in self._benchmark)
+                summary["benchmark_symbol"] = ",".join(f"{self._env.data_proxy.instrument(o).symbol if o not in self.NULL_OID else 'null'}:{w}" for o, w in self._benchmark)
 
         risk_free_rate = data_proxy.get_risk_free_rate(self._env.config.base.start_date, self._env.config.base.end_date)
         risk = Risk(

{rqalpha-5.4.3 → rqalpha-5.5.0/rqalpha.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rqalpha
-Version: 5.4.3
+Version: 5.5.0
 Summary: Ricequant Algorithm Trading System
 Home-page: https://github.com/ricequant/rqalpha
 Author: ricequant

{rqalpha-5.4.3 → rqalpha-5.5.0}/setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = rqalpha
-version = 5.4.3
+version = 5.5.0
 
 [versioneer]
 VCS = git

{rqalpha-5.4.3 → rqalpha-5.5.0}/versioneer.py

@@ -1665,13 +1665,9 @@ def render_pep440_ricequant(pieces):
     # # 如果是dev和master分支或者hotfix分支来的，或者是一个tag，那就用pep440的版本号，否则带上git commit id
     if tracking_branch in ["origin/develop", "origin/master"] or tracking_branch.startswith("origin/hotfix/") or pieces[
         "distance"] == 0:
-        if pieces["dirty"]:
-            rendered += ".dirty"
         return rendered
 
     rendered += "+%s" % format(pieces["short"])
-    if pieces["dirty"]:
-        rendered += ".dirty"
     return rendered