Deploying to main from @ pyt-team/TopoModelX@359d62e 🚀

Author: ffl096

topomodelx/.doctrees/environment.pickle

@@ -104,8 +104,6 @@
    "source": [
     "shrec, _ = tnx.datasets.shrec_16(size=\"small\")\n",
     "\n",
-    "shrec = {key: np.array(value) for key, value in shrec.items()}\n",
-    "\n",
     "x_0s = shrec[\"node_feat\"]\n",
     "x_1s = shrec[\"edge_feat\"]\n",
     "x_2s = shrec[\"face_feat\"]\n",
@@ -900,7 +898,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
+   "display_name": ".venv",
    "language": "python",
    "name": "python3"
   },
@@ -914,7 +912,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.3"
+   "version": "3.12.7"
   }
  },
  "nbformat": 4,

topomodelx/.doctrees/nbsphinx/notebooks/simplicial/scnn_train.ipynb

@@ -29,7 +29,7 @@
   <link href="../_static/styles/theme.css?digest=8878045cc6db502f8baf" rel="stylesheet" />
 <link href="../_static/styles/pydata-sphinx-theme.css?digest=8878045cc6db502f8baf" rel="stylesheet" />
 
-    <link rel="stylesheet" type="text/css" href="../_static/pygments.css?v=a746c00c" />
+    <link rel="stylesheet" type="text/css" href="../_static/pygments.css?v=8f2a1f02" />
     <link rel="stylesheet" type="text/css" href="../_static/copybutton.css?v=76b2166b" />
     <link rel="stylesheet" type="text/css" href="../_static/sg_gallery.css?v=d2d258e8" />
 

topomodelx/.doctrees/notebooks/cell/can_train.doctree

@@ -29,7 +29,7 @@
   <link href="../../../_static/styles/theme.css?digest=8878045cc6db502f8baf" rel="stylesheet" />
 <link href="../../../_static/styles/pydata-sphinx-theme.css?digest=8878045cc6db502f8baf" rel="stylesheet" />
 
-    <link rel="stylesheet" type="text/css" href="../../../_static/pygments.css?v=a746c00c" />
+    <link rel="stylesheet" type="text/css" href="../../../_static/pygments.css?v=8f2a1f02" />
     <link rel="stylesheet" type="text/css" href="../../../_static/copybutton.css?v=76b2166b" />
     <link rel="stylesheet" type="text/css" href="../../../_static/sg_gallery.css?v=d2d258e8" />
 
@@ -364,14 +364,14 @@
 
   <h1>Source code for topomodelx.base.aggregation</h1><div class="highlight"><pre>
 <span></span><span class="sd">&quot;&quot;&quot;Aggregation module.&quot;&quot;&quot;</span>
-<span class="kn">from</span> <span class="nn">typing</span> <span class="kn">import</span> <span class="n">Literal</span>
+<span class="kn">from</span><span class="w"> </span><span class="nn">typing</span><span class="w"> </span><span class="kn">import</span> <span class="n">Literal</span>
 
-<span class="kn">import</span> <span class="nn">torch</span>
+<span class="kn">import</span><span class="w"> </span><span class="nn">torch</span>
 
 
 <div class="viewcode-block" id="Aggregation">
 <a class="viewcode-back" href="../../../api/base/aggregation.html#topomodelx.base.aggregation.Aggregation">[docs]</a>
-<span class="k">class</span> <span class="nc">Aggregation</span><span class="p">(</span><span class="n">torch</span><span class="o">.</span><span class="n">nn</span><span class="o">.</span><span class="n">Module</span><span class="p">):</span>
+<span class="k">class</span><span class="w"> </span><span class="nc">Aggregation</span><span class="p">(</span><span class="n">torch</span><span class="o">.</span><span class="n">nn</span><span class="o">.</span><span class="n">Module</span><span class="p">):</span>
 <span class="w">    </span><span class="sd">&quot;&quot;&quot;Message passing layer.</span>
 
 <span class="sd">    Parameters</span>
@@ -382,7 +382,7 @@ <h1>Source code for topomodelx.base.aggregation</h1><div class="highlight"><pre>
 <span class="sd">        Update method to apply to merged message.</span>
 <span class="sd">    &quot;&quot;&quot;</span>
 
-    <span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span>
+    <span class="k">def</span><span class="w"> </span><span class="fm">__init__</span><span class="p">(</span>
         <span class="bp">self</span><span class="p">,</span>
         <span class="n">aggr_func</span><span class="p">:</span> <span class="n">Literal</span><span class="p">[</span><span class="s2">&quot;mean&quot;</span><span class="p">,</span> <span class="s2">&quot;sum&quot;</span><span class="p">]</span> <span class="o">=</span> <span class="s2">&quot;sum&quot;</span><span class="p">,</span>
         <span class="n">update_func</span><span class="p">:</span> <span class="n">Literal</span><span class="p">[</span><span class="s2">&quot;relu&quot;</span><span class="p">,</span> <span class="s2">&quot;sigmoid&quot;</span><span class="p">,</span> <span class="s2">&quot;tanh&quot;</span><span class="p">]</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="s2">&quot;sigmoid&quot;</span><span class="p">,</span>
@@ -393,7 +393,7 @@ <h1>Source code for topomodelx.base.aggregation</h1><div class="highlight"><pre>
 
 <div class="viewcode-block" id="Aggregation.update">
 <a class="viewcode-back" href="../../../api/base/aggregation.html#topomodelx.base.aggregation.Aggregation.update">[docs]</a>
-    <span class="k">def</span> <span class="nf">update</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">inputs</span><span class="p">):</span>
+    <span class="k">def</span><span class="w"> </span><span class="nf">update</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">inputs</span><span class="p">):</span>
 <span class="w">        </span><span class="sd">&quot;&quot;&quot;Update (Step 4).</span>
 
 <span class="sd">        Parameters</span>
@@ -417,7 +417,7 @@ <h1>Source code for topomodelx.base.aggregation</h1><div class="highlight"><pre>
 
 <div class="viewcode-block" id="Aggregation.forward">
 <a class="viewcode-back" href="../../../api/base/aggregation.html#topomodelx.base.aggregation.Aggregation.forward">[docs]</a>
-    <span class="k">def</span> <span class="nf">forward</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">x</span><span class="p">):</span>
+    <span class="k">def</span><span class="w"> </span><span class="nf">forward</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">x</span><span class="p">):</span>
 <span class="w">        </span><span class="sd">&quot;&quot;&quot;Forward pass.</span>
 
 <span class="sd">        Parameters</span>

topomodelx/.doctrees/notebooks/simplicial/san_train.doctree

@@ -29,7 +29,7 @@
   <link href="../../../_static/styles/theme.css?digest=8878045cc6db502f8baf" rel="stylesheet" />
 <link href="../../../_static/styles/pydata-sphinx-theme.css?digest=8878045cc6db502f8baf" rel="stylesheet" />
 
-    <link rel="stylesheet" type="text/css" href="../../../_static/pygments.css?v=a746c00c" />
+    <link rel="stylesheet" type="text/css" href="../../../_static/pygments.css?v=8f2a1f02" />
     <link rel="stylesheet" type="text/css" href="../../../_static/copybutton.css?v=76b2166b" />
     <link rel="stylesheet" type="text/css" href="../../../_static/sg_gallery.css?v=d2d258e8" />
 
@@ -364,17 +364,17 @@
 
   <h1>Source code for topomodelx.base.conv</h1><div class="highlight"><pre>
 <span></span><span class="sd">&quot;&quot;&quot;Convolutional layer for message passing.&quot;&quot;&quot;</span>
-<span class="kn">from</span> <span class="nn">typing</span> <span class="kn">import</span> <span class="n">Literal</span>
+<span class="kn">from</span><span class="w"> </span><span class="nn">typing</span><span class="w"> </span><span class="kn">import</span> <span class="n">Literal</span>
 
-<span class="kn">import</span> <span class="nn">torch</span>
-<span class="kn">from</span> <span class="nn">torch.nn.parameter</span> <span class="kn">import</span> <span class="n">Parameter</span>
+<span class="kn">import</span><span class="w"> </span><span class="nn">torch</span>
+<span class="kn">from</span><span class="w"> </span><span class="nn">torch.nn.parameter</span><span class="w"> </span><span class="kn">import</span> <span class="n">Parameter</span>
 
-<span class="kn">from</span> <span class="nn">topomodelx.base.message_passing</span> <span class="kn">import</span> <span class="n">MessagePassing</span>
+<span class="kn">from</span><span class="w"> </span><span class="nn">topomodelx.base.message_passing</span><span class="w"> </span><span class="kn">import</span> <span class="n">MessagePassing</span>
 
 
 <div class="viewcode-block" id="Conv">
 <a class="viewcode-back" href="../../../api/base/conv.html#topomodelx.base.conv.Conv">[docs]</a>
-<span class="k">class</span> <span class="nc">Conv</span><span class="p">(</span><span class="n">MessagePassing</span><span class="p">):</span>
+<span class="k">class</span><span class="w"> </span><span class="nc">Conv</span><span class="p">(</span><span class="n">MessagePassing</span><span class="p">):</span>
 <span class="w">    </span><span class="sd">&quot;&quot;&quot;Message passing: steps 1, 2, and 3.</span>
 
 <span class="sd">    Builds the message passing route given by one neighborhood matrix.</span>
@@ -401,7 +401,7 @@ <h1>Source code for topomodelx.base.conv</h1><div class="highlight"><pre>
 <span class="sd">        NB: if `False` in_channels has to be equal to out_channels.</span>
 <span class="sd">    &quot;&quot;&quot;</span>
 
-    <span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span>
+    <span class="k">def</span><span class="w"> </span><span class="fm">__init__</span><span class="p">(</span>
         <span class="bp">self</span><span class="p">,</span>
         <span class="n">in_channels</span><span class="p">,</span>
         <span class="n">out_channels</span><span class="p">,</span>
@@ -443,7 +443,7 @@ <h1>Source code for topomodelx.base.conv</h1><div class="highlight"><pre>
 
 <div class="viewcode-block" id="Conv.update">
 <a class="viewcode-back" href="../../../api/base/conv.html#topomodelx.base.conv.Conv.update">[docs]</a>
-    <span class="k">def</span> <span class="nf">update</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">x_message_on_target</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">:</span>
+    <span class="k">def</span><span class="w"> </span><span class="nf">update</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">x_message_on_target</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">:</span>
 <span class="w">        </span><span class="sd">&quot;&quot;&quot;Update embeddings on each cell (step 4).</span>
 
 <span class="sd">        Parameters</span>
@@ -465,7 +465,7 @@ <h1>Source code for topomodelx.base.conv</h1><div class="highlight"><pre>
 
 <div class="viewcode-block" id="Conv.forward">
 <a class="viewcode-back" href="../../../api/base/conv.html#topomodelx.base.conv.Conv.forward">[docs]</a>
-    <span class="k">def</span> <span class="nf">forward</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">x_source</span><span class="p">,</span> <span class="n">neighborhood</span><span class="p">,</span> <span class="n">x_target</span><span class="o">=</span><span class="kc">None</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">:</span>
+    <span class="k">def</span><span class="w"> </span><span class="nf">forward</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">x_source</span><span class="p">,</span> <span class="n">neighborhood</span><span class="p">,</span> <span class="n">x_target</span><span class="o">=</span><span class="kc">None</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">:</span>
 <span class="w">        </span><span class="sd">&quot;&quot;&quot;Forward pass.</span>
 
 <span class="sd">        This implements message passing:</span>