|
5 | 5 | <head> |
6 | 6 | <meta charset="utf-8" /> |
7 | 7 | <meta name="viewport" content="width=device-width, initial-scale=1.0" /> |
8 | | - <title>cmdstanpy.stanfit.runset — CmdStanPy 1.0.7 documentation</title> |
| 8 | + <title>cmdstanpy.stanfit.runset — CmdStanPy 1.0.8 documentation</title> |
9 | 9 |
|
10 | 10 | <!-- Loaded before other Sphinx assets --> |
11 | 11 | <link href="../../../_static/styles/theme.css?digest=1999514e3f237ded88cf" rel="stylesheet"> |
|
56 | 56 | <div id="navbar-start"> |
57 | 57 |
|
58 | 58 | <!-- This will display the version of the docs --> |
59 | | -<a class='navbar-brand' href='../../../index.html'>CmdStanPy 1.0.7</a> |
| 59 | +<a class='navbar-brand' href='../../../index.html'>CmdStanPy 1.0.8</a> |
60 | 60 |
|
61 | 61 | </div> |
62 | 62 |
|
@@ -213,6 +213,7 @@ <h1>Source code for cmdstanpy.stanfit.runset</h1><div class="highlight"><pre> |
213 | 213 | <span class="k">else</span><span class="p">:</span> |
214 | 214 | <span class="bp">self</span><span class="o">.</span><span class="n">_num_procs</span> <span class="o">=</span> <span class="mi">1</span> |
215 | 215 | <span class="bp">self</span><span class="o">.</span><span class="n">_retcodes</span> <span class="o">=</span> <span class="p">[</span><span class="o">-</span><span class="mi">1</span> <span class="k">for</span> <span class="n">_</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_num_procs</span><span class="p">)]</span> |
| 216 | + <span class="bp">self</span><span class="o">.</span><span class="n">_timeout_flags</span> <span class="o">=</span> <span class="p">[</span><span class="kc">False</span> <span class="k">for</span> <span class="n">_</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_num_procs</span><span class="p">)]</span> |
216 | 217 | <span class="k">if</span> <span class="n">chain_ids</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span> |
217 | 218 | <span class="n">chain_ids</span> <span class="o">=</span> <span class="p">[</span><span class="n">i</span> <span class="o">+</span> <span class="mi">1</span> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="n">chains</span><span class="p">)]</span> |
218 | 219 | <span class="bp">self</span><span class="o">.</span><span class="n">_chain_ids</span> <span class="o">=</span> <span class="n">chain_ids</span> |
@@ -399,6 +400,10 @@ <h1>Source code for cmdstanpy.stanfit.runset</h1><div class="highlight"><pre> |
399 | 400 | <span class="sd">"""Set retcode at process[idx] to val."""</span> |
400 | 401 | <span class="bp">self</span><span class="o">.</span><span class="n">_retcodes</span><span class="p">[</span><span class="n">idx</span><span class="p">]</span> <span class="o">=</span> <span class="n">val</span> |
401 | 402 |
|
| 403 | + <span class="k">def</span> <span class="nf">_set_timeout_flag</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">idx</span><span class="p">:</span> <span class="nb">int</span><span class="p">,</span> <span class="n">val</span><span class="p">:</span> <span class="nb">bool</span><span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span> |
| 404 | + <span class="sd">"""Set timeout_flag at process[idx] to val."""</span> |
| 405 | + <span class="bp">self</span><span class="o">.</span><span class="n">_timeout_flags</span><span class="p">[</span><span class="n">idx</span><span class="p">]</span> <span class="o">=</span> <span class="n">val</span> |
| 406 | + |
402 | 407 | <div class="viewcode-block" id="RunSet.get_err_msgs"><a class="viewcode-back" href="../../../internal_api.html#cmdstanpy.stanfit.RunSet.get_err_msgs">[docs]</a> <span class="k">def</span> <span class="nf">get_err_msgs</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> |
403 | 408 | <span class="sd">"""Checks console messages for each CmdStan run."""</span> |
404 | 409 | <span class="n">msgs</span> <span class="o">=</span> <span class="p">[]</span> |
@@ -462,7 +467,14 @@ <h1>Source code for cmdstanpy.stanfit.runset</h1><div class="highlight"><pre> |
462 | 467 | <span class="k">except</span> <span class="p">(</span><span class="ne">IOError</span><span class="p">,</span> <span class="ne">OSError</span><span class="p">,</span> <span class="ne">PermissionError</span><span class="p">)</span> <span class="k">as</span> <span class="n">e</span><span class="p">:</span> |
463 | 468 | <span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span> |
464 | 469 | <span class="s1">'Cannot save to file: </span><span class="si">{}</span><span class="s1">'</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">to_path</span><span class="p">)</span> |
465 | | - <span class="p">)</span> <span class="kn">from</span> <span class="nn">e</span></div></div> |
| 470 | + <span class="p">)</span> <span class="kn">from</span> <span class="nn">e</span></div> |
| 471 | + |
| 472 | + <span class="k">def</span> <span class="nf">raise_for_timeouts</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span> |
| 473 | + <span class="k">if</span> <span class="nb">any</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_timeout_flags</span><span class="p">):</span> |
| 474 | + <span class="k">raise</span> <span class="ne">TimeoutError</span><span class="p">(</span> |
| 475 | + <span class="sa">f</span><span class="s2">"</span><span class="si">{</span><span class="nb">sum</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_timeout_flags</span><span class="p">)</span><span class="si">}</span><span class="s2"> of </span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">num_procs</span><span class="si">}</span><span class="s2"> processes "</span> |
| 476 | + <span class="s2">"timed out"</span> |
| 477 | + <span class="p">)</span></div> |
466 | 478 | </pre></div> |
467 | 479 |
|
468 | 480 | </div> |
|
![github-actions[bot]](https://avatars.githubusercontent.com/in/15368?v=4&size=40){kind=avatar}
0 commit comments