Skip to content

Commit

Permalink
deploy: 42af3a9
Browse files Browse the repository at this point in the history
mieskolainen committed Aug 1, 2024

Verified

This commit was signed with the committer’s verified signature.
chenchanglew Zac Lew
1 parent b9893f9 commit f91aac6
Showing 2 changed files with 6 additions and 6 deletions.
8 changes: 4 additions & 4 deletions _modules/icenet/tools/aux.html
Original file line number Diff line number Diff line change
@@ -583,15 +583,15 @@ <h1>Source code for icenet.tools.aux</h1><div class="highlight"><pre>

<div class="viewcode-block" id="recursive_concatenate">
<a class="viewcode-back" href="../../../modules/icenet.html#icenet.tools.aux.recursive_concatenate">[docs]</a>
<span class="k">def</span> <span class="nf">recursive_concatenate</span><span class="p">(</span><span class="n">array_list</span><span class="p">,</span> <span class="n">axis</span><span class="p">:</span> <span class="nb">int</span><span class="o">=</span><span class="mi">0</span><span class="p">,</span> <span class="n">max_batch_size</span><span class="p">:</span> <span class="nb">int</span> <span class="o">=</span> <span class="mi">32</span><span class="p">):</span>
<span class="k">def</span> <span class="nf">recursive_concatenate</span><span class="p">(</span><span class="n">array_list</span><span class="p">,</span> <span class="n">max_batch_size</span><span class="p">:</span> <span class="nb">int</span><span class="o">=</span><span class="mi">32</span><span class="p">,</span> <span class="n">axis</span><span class="p">:</span> <span class="nb">int</span><span class="o">=</span><span class="mi">0</span><span class="p">):</span>
<span class="w"> </span><span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Concatenate a list of arrays in a recursive way</span>
<span class="sd"> (to avoid possible problems with one big concatenation e.g. with Awkward)</span>
<span class="sd"> </span>
<span class="sd"> Args:</span>
<span class="sd"> array_list: a list of Awkward or Numpy arrays</span>
<span class="sd"> axis: axis to concatenate over</span>
<span class="sd"> max_batch_size: maximum number of list elements per concatenation</span>
<span class="sd"> axis: axis to concatenate over</span>
<span class="sd"> </span>
<span class="sd"> Returns:</span>
<span class="sd"> concatenated array</span>
@@ -613,8 +613,8 @@ <h1>Source code for icenet.tools.aux</h1><div class="highlight"><pre>
<span class="c1"># Split the list into two halves and recursively concatenate each half</span>
<span class="k">else</span><span class="p">:</span>
<span class="n">mid</span> <span class="o">=</span> <span class="p">(</span><span class="n">n</span> <span class="o">+</span> <span class="mi">1</span><span class="p">)</span> <span class="o">//</span> <span class="mi">2</span> <span class="c1"># handle odd length</span>
<span class="n">left</span> <span class="o">=</span> <span class="n">recursive_concatenate</span><span class="p">(</span><span class="n">array_list</span><span class="p">[:</span><span class="n">mid</span><span class="p">],</span> <span class="n">max_batch_size</span><span class="p">)</span>
<span class="n">right</span> <span class="o">=</span> <span class="n">recursive_concatenate</span><span class="p">(</span><span class="n">array_list</span><span class="p">[</span><span class="n">mid</span><span class="p">:],</span> <span class="n">max_batch_size</span><span class="p">)</span>
<span class="n">left</span> <span class="o">=</span> <span class="n">recursive_concatenate</span><span class="p">(</span><span class="n">array_list</span><span class="p">[:</span><span class="n">mid</span><span class="p">],</span> <span class="n">max_bath_size</span><span class="o">=</span><span class="n">max_batch_size</span><span class="p">,</span> <span class="n">axis</span><span class="o">=</span><span class="n">axis</span><span class="p">)</span>
<span class="n">right</span> <span class="o">=</span> <span class="n">recursive_concatenate</span><span class="p">(</span><span class="n">array_list</span><span class="p">[</span><span class="n">mid</span><span class="p">:],</span> <span class="n">max_bath_size</span><span class="o">=</span><span class="n">max_batch_size</span><span class="p">,</span> <span class="n">axis</span><span class="o">=</span><span class="n">axis</span><span class="p">)</span>
<span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">left</span><span class="p">,</span> <span class="n">ak</span><span class="o">.</span><span class="n">Array</span><span class="p">)</span> <span class="ow">or</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">right</span><span class="p">,</span> <span class="n">ak</span><span class="o">.</span><span class="n">Array</span><span class="p">):</span>
<span class="k">return</span> <span class="n">ak</span><span class="o">.</span><span class="n">concatenate</span><span class="p">([</span><span class="n">left</span><span class="p">,</span> <span class="n">right</span><span class="p">],</span> <span class="n">axis</span><span class="o">=</span><span class="n">axis</span><span class="p">)</span>
<span class="k">else</span><span class="p">:</span>
4 changes: 2 additions & 2 deletions modules/icenet.html
Original file line number Diff line number Diff line change
@@ -3336,15 +3336,15 @@ <h2>icenet.tools<a class="headerlink" href="#icenet-tools" title="Link to this h

<dl class="py function">
<dt class="sig sig-object py" id="icenet.tools.aux.recursive_concatenate">
<span class="sig-name descname"><span class="pre">recursive_concatenate</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">array_list</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">axis</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/functions.html#int" title="(in Python v3.12)"><span class="pre">int</span></a></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">0</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">max_batch_size</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/functions.html#int" title="(in Python v3.12)"><span class="pre">int</span></a></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">32</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/icenet/tools/aux.html#recursive_concatenate"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#icenet.tools.aux.recursive_concatenate" title="Link to this definition">¶</a></dt>
<span class="sig-name descname"><span class="pre">recursive_concatenate</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">array_list</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">max_batch_size</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/functions.html#int" title="(in Python v3.12)"><span class="pre">int</span></a></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">32</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">axis</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/functions.html#int" title="(in Python v3.12)"><span class="pre">int</span></a></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">0</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/icenet/tools/aux.html#recursive_concatenate"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#icenet.tools.aux.recursive_concatenate" title="Link to this definition">¶</a></dt>
<dd><p>Concatenate a list of arrays in a recursive way
(to avoid possible problems with one big concatenation e.g. with Awkward)</p>
<dl class="field-list simple">
<dt class="field-odd">Parameters<span class="colon">:</span></dt>
<dd class="field-odd"><ul class="simple">
<li><p><strong>array_list</strong> – a list of Awkward or Numpy arrays</p></li>
<li><p><strong>axis</strong> – axis to concatenate over</p></li>
<li><p><strong>max_batch_size</strong> – maximum number of list elements per concatenation</p></li>
<li><p><strong>axis</strong> – axis to concatenate over</p></li>
</ul>
</dd>
<dt class="field-even">Returns<span class="colon">:</span></dt>

0 comments on commit f91aac6

Please sign in to comment.