Skip to content

Commit

Permalink
Deploying to gh-pages from @ c87dfbf 🚀
Browse files Browse the repository at this point in the history
  • Loading branch information
asogaard committed Mar 18, 2024
1 parent c9dd4c1 commit 3814965
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions _modules/graphnet/models/graphs/nodes/nodes.html
Original file line number Diff line number Diff line change
Expand Up @@ -613,8 +613,6 @@ <h1 id="modules-graphnet-models-graphs-nodes-nodes--page-root">Source code for g
<span class="w"> </span><span class="sd">"""Construct nodes from raw node features ´x´."""</span>
<span class="c1"># Cast to Numpy</span>
<span class="n">x</span> <span class="o">=</span> <span class="n">x</span><span class="o">.</span><span class="n">numpy</span><span class="p">()</span>
<span class="k">if</span> <span class="n">x</span><span class="o">.</span><span class="n">shape</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span> <span class="o">==</span> <span class="mi">0</span><span class="p">:</span>
<span class="k">return</span> <span class="n">Data</span><span class="p">(</span><span class="n">x</span><span class="o">=</span><span class="n">torch</span><span class="o">.</span><span class="n">tensor</span><span class="p">(</span><span class="n">np</span><span class="o">.</span><span class="n">column_stack</span><span class="p">([</span><span class="n">x</span><span class="p">,</span> <span class="p">[]])))</span>
<span class="c1"># if there is no charge column add a dummy column of zeros with the same shape as the time column</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">_charge_index</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="n">charge_index</span><span class="p">:</span> <span class="nb">int</span> <span class="o">=</span> <span class="nb">len</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_keys</span><span class="p">)</span>
Expand Down

0 comments on commit 3814965

Please sign in to comment.