win下如果只开一个进程，那么只在主进程里运行，方便pdb debug且与joblib运行方式保持一致

ngwangshu · Sep 1, 2017 · ab769f7 · ab769f7
1 parent 37a5591
commit ab769f7
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 6 deletions.
diff --git a/abupy/CoreBu/ABuParallel.py b/abupy/CoreBu/ABuParallel.py
@@ -80,11 +80,16 @@ def when_done(r):
  # 主要为了适配 n_jobs = -1，joblib中启动cpu个数个进程并行执行
  self.n_jobs = ABuEnv.g_cpu_cnt
 
- with ProcessPoolExecutor(max_workers=self.n_jobs) as pool:
+ if self.n_jobs == 1:
+ # 如果只开一个进程，那么只在主进程里运行，方便pdb debug且与joblib运行方式保持一致
  for jb in iterable:
- # 这里iterable里每一个元素是delayed.delayed_function保留的tuple
- future_result = pool.submit(jb[0], *jb[1], **jb[2])
- future_result.add_done_callback(when_done)
+ result.append(jb[0](*jb[1], **jb[2]))
+ else:
+ with ProcessPoolExecutor(max_workers=self.n_jobs) as pool:
+ for jb in iterable:
+ # 这里iterable里每一个元素是delayed.delayed_function保留的tuple
+ future_result = pool.submit(jb[0], *jb[1], **jb[2])
+ future_result.add_done_callback(when_done)
  return result
 
 

diff --git a/abupy_lecture/16-UMP主裁交易决策(ABU量化使用文档).ipynb b/abupy_lecture/16-UMP主裁交易决策(ABU量化使用文档).ipynb
@@ -131,7 +131,7 @@
  "cell_type": "markdown",
  "metadata": {},
  "source": [
- "使用load_abu_result_tuple读取上一节保存在本地的训练集数据："
+ "在运行完成第15节中相关内容后，使用load_abu_result_tuple读取上一节保存在本地的训练集数据："
  ]
  },
  {

diff --git a/abupy_lecture/17-UMP边裁交易决策(ABU量化使用文档).ipynb b/abupy_lecture/17-UMP边裁交易决策(ABU量化使用文档).ipynb
@@ -126,7 +126,7 @@
  "cell_type": "markdown",
  "metadata": {},
  "source": [
- "使用load_abu_result_tuple读取第15节中保存在本地的训练集数据："
+ "在运行完成第15节中相关内容后，使用load_abu_result_tuple读取第15节中保存在本地的训练集数据："
  ]
  },
  {