Github action: auto-update.

neuraloperator · Jan 14, 2025 · 4521054 · 4521054
1 parent 98f91ed
commit 4521054
Show file tree

Hide file tree

Showing 118 changed files with 859 additions and 849 deletions.
diff --git a/dev/_downloads/07fcc19ba03226cd3d83d4e40ec44385/auto_examples_python.zip b/dev/_downloads/07fcc19ba03226cd3d83d4e40ec44385/auto_examples_python.zip
diff --git a/dev/_downloads/0d78e075dd52a34e158d7f5f710dfe89/plot_incremental_FNO_darcy.zip b/dev/_downloads/0d78e075dd52a34e158d7f5f710dfe89/plot_incremental_FNO_darcy.zip
diff --git a/dev/_downloads/20c43dd37baf603889c4dc23e93bdb60/plot_count_flops.zip b/dev/_downloads/20c43dd37baf603889c4dc23e93bdb60/plot_count_flops.zip
diff --git a/dev/_downloads/3864a2d85c7ce11adeac9580559229ab/plot_darcy_flow.zip b/dev/_downloads/3864a2d85c7ce11adeac9580559229ab/plot_darcy_flow.zip
diff --git a/dev/_downloads/3faf9d2eaee5cc8e9f1c631c002ce544/plot_darcy_flow_spectrum.zip b/dev/_downloads/3faf9d2eaee5cc8e9f1c631c002ce544/plot_darcy_flow_spectrum.zip
diff --git a/dev/_downloads/4e1cba46bb51062a073424e1efbaad57/plot_DISCO_convolutions.zip b/dev/_downloads/4e1cba46bb51062a073424e1efbaad57/plot_DISCO_convolutions.zip
diff --git a/dev/_downloads/5e60095ce99919773daa83384f767e02/plot_SFNO_swe.zip b/dev/_downloads/5e60095ce99919773daa83384f767e02/plot_SFNO_swe.zip
diff --git a/dev/_downloads/645da00b8fbbb9bb5cae877fd0f31635/plot_FNO_darcy.zip b/dev/_downloads/645da00b8fbbb9bb5cae877fd0f31635/plot_FNO_darcy.zip
diff --git a/dev/_downloads/6f1e7a639e0699d6164445b55e6c116d/auto_examples_jupyter.zip b/dev/_downloads/6f1e7a639e0699d6164445b55e6c116d/auto_examples_jupyter.zip
diff --git a/dev/_downloads/7296405f6df7c2cfe184e9b258cee33e/checkpoint_FNO_darcy.zip b/dev/_downloads/7296405f6df7c2cfe184e9b258cee33e/checkpoint_FNO_darcy.zip
diff --git a/dev/_downloads/cefc537c5730a6b3e916b83c1fd313d6/plot_UNO_darcy.zip b/dev/_downloads/cefc537c5730a6b3e916b83c1fd313d6/plot_UNO_darcy.zip
diff --git a/dev/_images/sphx_glr_plot_SFNO_swe_001.png b/dev/_images/sphx_glr_plot_SFNO_swe_001.png
diff --git a/dev/_images/sphx_glr_plot_SFNO_swe_thumb.png b/dev/_images/sphx_glr_plot_SFNO_swe_thumb.png
diff --git a/dev/_images/sphx_glr_plot_UNO_darcy_001.png b/dev/_images/sphx_glr_plot_UNO_darcy_001.png
diff --git a/dev/_images/sphx_glr_plot_UNO_darcy_thumb.png b/dev/_images/sphx_glr_plot_UNO_darcy_thumb.png
diff --git a/dev/_modules/index.html b/dev/_modules/index.html
@@ -14,7 +14,7 @@
 <link rel="shortcut icon" href="../_static/favicon/favicon.ico">
 <meta name="theme-color" content="#ffffff">
 
-    <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/tensorly_style.css?v=a02e9698" />
     <link rel="stylesheet" type="text/css" href="../_static/sg_gallery.css?v=d2d258e8" />
     <link rel="stylesheet" type="text/css" href="../_static/sg_gallery-binder.css?v=f4aeca0c" />

diff --git a/dev/_modules/neuralop/data/datasets/car_cfd_dataset.html b/dev/_modules/neuralop/data/datasets/car_cfd_dataset.html
@@ -14,7 +14,7 @@
 <link rel="shortcut icon" href="../../../../_static/favicon/favicon.ico">
 <meta name="theme-color" content="#ffffff">
 
-    <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/tensorly_style.css?v=a02e9698" />
     <link rel="stylesheet" type="text/css" href="../../../../_static/sg_gallery.css?v=d2d258e8" />
     <link rel="stylesheet" type="text/css" href="../../../../_static/sg_gallery-binder.css?v=f4aeca0c" />
@@ -113,15 +113,17 @@
       <div class="container content main-content">
 
   <h1>Source code for neuralop.data.datasets.car_cfd_dataset</h1><div class="highlight"><pre>
-<span></span><span class="kn">from</span> <span class="nn">typing</span> <span class="kn">import</span> <span class="n">List</span><span class="p">,</span> <span class="n">Union</span>
-<span class="kn">from</span> <span class="nn">pathlib</span> <span class="kn">import</span> <span class="n">Path</span>
+<span></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">List</span><span class="p">,</span> <span class="n">Union</span>
+<span class="kn">from</span><span class="w"> </span><span class="nn">pathlib</span><span class="w"> </span><span class="kn">import</span> <span class="n">Path</span>
 
-<span class="kn">from</span> <span class="nn">.mesh_datamodule</span> <span class="kn">import</span> <span class="n">MeshDataModule</span>
-<span class="kn">from</span> <span class="nn">.web_utils</span> <span class="kn">import</span> <span class="n">download_from_zenodo_record</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">.mesh_datamodule</span><span class="w"> </span><span class="kn">import</span> <span class="n">MeshDataModule</span>
+<span class="kn">from</span><span class="w"> </span><span class="nn">.web_utils</span><span class="w"> </span><span class="kn">import</span> <span class="n">download_from_zenodo_record</span>
 
 <div class="viewcode-block" id="CarCFDDataset">
 <a class="viewcode-back" href="../../../../modules/generated/neuralop.data.datasets.car_cfd_dataset.CarCFDDataset.html#neuralop.data.datasets.car_cfd_dataset.CarCFDDataset">[docs]</a>
-<span class="k">class</span> <span class="nc">CarCFDDataset</span><span class="p">(</span><span class="n">MeshDataModule</span><span class="p">):</span>
+<span class="k">class</span><span class="w"> </span><span class="nc">CarCFDDataset</span><span class="p">(</span><span class="n">MeshDataModule</span><span class="p">):</span>
 <span class="w">    </span><span class="sd">&quot;&quot;&quot;CarCFDDataset is a processed version of the dataset introduced in</span>
 <span class="sd">    [1]_, which encodes a triangular mesh over the surface of a 3D model car</span>
 <span class="sd">    and provides the air pressure at each centroid and vertex of the mesh when</span>
@@ -171,7 +173,7 @@ <h1>Source code for neuralop.data.datasets.car_cfd_dataset</h1><div class="highl
 <span class="sd">        https://dl.acm.org/doi/10.1145/3197517.3201325.</span>
 <span class="sd">    &quot;&quot;&quot;</span>
 
-    <span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span><span class="bp">self</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">root_dir</span><span class="p">:</span> <span class="n">Union</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="n">Path</span><span class="p">],</span>
         <span class="n">n_train</span><span class="p">:</span> <span class="nb">int</span> <span class="o">=</span> <span class="mi">1</span><span class="p">,</span>
         <span class="n">n_test</span><span class="p">:</span> <span class="nb">int</span> <span class="o">=</span> <span class="mi">1</span><span class="p">,</span>
@@ -198,7 +200,15 @@ <h1>Source code for neuralop.data.datasets.car_cfd_dataset</h1><div class="highl
             <span class="n">n_test</span><span class="o">=</span><span class="n">n_test</span><span class="p">,</span>
             <span class="n">query_res</span><span class="o">=</span><span class="n">query_res</span><span class="p">,</span>
             <span class="n">attributes</span><span class="o">=</span><span class="p">[</span><span class="s1">&#39;press&#39;</span><span class="p">]</span>
-        <span class="p">)</span></div>
+        <span class="p">)</span>
+
+        <span class="c1"># process data list to remove specific vertices from pressure to match number of vertices</span>
+        <span class="k">for</span> <span class="n">i</span><span class="p">,</span> <span class="n">data</span> <span class="ow">in</span> <span class="nb">enumerate</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">train_data</span><span class="o">.</span><span class="n">data_list</span><span class="p">):</span>
+            <span class="n">press</span> <span class="o">=</span> <span class="n">data</span><span class="p">[</span><span class="s1">&#39;press&#39;</span><span class="p">]</span>
+            <span class="bp">self</span><span class="o">.</span><span class="n">train_data</span><span class="o">.</span><span class="n">data_list</span><span class="p">[</span><span class="n">i</span><span class="p">][</span><span class="s1">&#39;press&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</span><span class="n">cat</span><span class="p">((</span><span class="n">press</span><span class="p">[:,</span><span class="mi">0</span><span class="p">:</span><span class="mi">16</span><span class="p">],</span> <span class="n">press</span><span class="p">[:,</span><span class="mi">112</span><span class="p">:]),</span> <span class="n">axis</span><span class="o">=</span><span class="mi">1</span><span class="p">)</span>
+        <span class="k">for</span> <span class="n">i</span><span class="p">,</span> <span class="n">data</span> <span class="ow">in</span> <span class="nb">enumerate</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">test_data</span><span class="o">.</span><span class="n">data_list</span><span class="p">):</span>
+            <span class="n">press</span> <span class="o">=</span> <span class="n">data</span><span class="p">[</span><span class="s1">&#39;press&#39;</span><span class="p">]</span>
+            <span class="bp">self</span><span class="o">.</span><span class="n">test_data</span><span class="o">.</span><span class="n">data_list</span><span class="p">[</span><span class="n">i</span><span class="p">][</span><span class="s1">&#39;press&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</span><span class="n">cat</span><span class="p">((</span><span class="n">press</span><span class="p">[:,</span><span class="mi">0</span><span class="p">:</span><span class="mi">16</span><span class="p">],</span> <span class="n">press</span><span class="p">[:,</span><span class="mi">112</span><span class="p">:]),</span> <span class="n">axis</span><span class="o">=</span><span class="mi">1</span><span class="p">)</span></div>
 
 </pre></div>
 

diff --git a/dev/_modules/neuralop/data/datasets/darcy.html b/dev/_modules/neuralop/data/datasets/darcy.html
@@ -14,7 +14,7 @@
 <link rel="shortcut icon" href="../../../../_static/favicon/favicon.ico">
 <meta name="theme-color" content="#ffffff">
 
-    <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/tensorly_style.css?v=a02e9698" />
     <link rel="stylesheet" type="text/css" href="../../../../_static/sg_gallery.css?v=d2d258e8" />
     <link rel="stylesheet" type="text/css" href="../../../../_static/sg_gallery-binder.css?v=f4aeca0c" />
@@ -113,23 +113,23 @@
       <div class="container content main-content">
 
   <h1>Source code for neuralop.data.datasets.darcy</h1><div class="highlight"><pre>
-<span></span><span class="kn">import</span> <span class="nn">logging</span>
-<span class="kn">import</span> <span class="nn">os</span>
-<span class="kn">from</span> <span class="nn">pathlib</span> <span class="kn">import</span> <span class="n">Path</span>
-<span class="kn">from</span> <span class="nn">typing</span> <span class="kn">import</span> <span class="n">Union</span><span class="p">,</span> <span class="n">List</span>
+<span></span><span class="kn">import</span><span class="w"> </span><span class="nn">logging</span>
+<span class="kn">import</span><span class="w"> </span><span class="nn">os</span>
+<span class="kn">from</span><span class="w"> </span><span class="nn">pathlib</span><span class="w"> </span><span class="kn">import</span> <span class="n">Path</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">Union</span><span class="p">,</span> <span class="n">List</span>
 
-<span class="kn">from</span> <span class="nn">torch.utils.data</span> <span class="kn">import</span> <span class="n">DataLoader</span>
+<span class="kn">from</span><span class="w"> </span><span class="nn">torch.utils.data</span><span class="w"> </span><span class="kn">import</span> <span class="n">DataLoader</span>
 
-<span class="kn">from</span> <span class="nn">.pt_dataset</span> <span class="kn">import</span> <span class="n">PTDataset</span>
-<span class="kn">from</span> <span class="nn">.web_utils</span> <span class="kn">import</span> <span class="n">download_from_zenodo_record</span>
+<span class="kn">from</span><span class="w"> </span><span class="nn">.pt_dataset</span><span class="w"> </span><span class="kn">import</span> <span class="n">PTDataset</span>
+<span class="kn">from</span><span class="w"> </span><span class="nn">.web_utils</span><span class="w"> </span><span class="kn">import</span> <span class="n">download_from_zenodo_record</span>
 
-<span class="kn">from</span> <span class="nn">neuralop.utils</span> <span class="kn">import</span> <span class="n">get_project_root</span>
+<span class="kn">from</span><span class="w"> </span><span class="nn">neuralop.utils</span><span class="w"> </span><span class="kn">import</span> <span class="n">get_project_root</span>
 
 <span class="n">logger</span> <span class="o">=</span> <span class="n">logging</span><span class="o">.</span><span class="n">Logger</span><span class="p">(</span><span class="n">logging</span><span class="o">.</span><span class="n">root</span><span class="o">.</span><span class="n">level</span><span class="p">)</span>
 
 <div class="viewcode-block" id="DarcyDataset">
 <a class="viewcode-back" href="../../../../modules/generated/neuralop.data.datasets.darcy.DarcyDataset.html#neuralop.data.datasets.darcy.DarcyDataset">[docs]</a>
-<span class="k">class</span> <span class="nc">DarcyDataset</span><span class="p">(</span><span class="n">PTDataset</span><span class="p">):</span>
+<span class="k">class</span><span class="w"> </span><span class="nc">DarcyDataset</span><span class="p">(</span><span class="n">PTDataset</span><span class="p">):</span>
 <span class="w">    </span><span class="sd">&quot;&quot;&quot;</span>
 <span class="sd">    DarcyDataset stores data generated according to Darcy&#39;s Law.</span>
 <span class="sd">    Input is a coefficient function and outputs describe flow. </span>
@@ -143,7 +143,7 @@ <h1>Source code for neuralop.data.datasets.darcy</h1><div class="highlight"><pre
 <span class="sd">    data_processor: neuralop.data.transforms.DataProcessor to process data examples</span>
 <span class="sd">        optional, default is None</span>
 <span class="sd">    &quot;&quot;&quot;</span>
-    <span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span><span class="bp">self</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">root_dir</span><span class="p">:</span> <span class="n">Union</span><span class="p">[</span><span class="n">Path</span><span class="p">,</span> <span class="nb">str</span><span class="p">],</span>
                  <span class="n">n_train</span><span class="p">:</span> <span class="nb">int</span><span class="p">,</span>
                  <span class="n">n_tests</span><span class="p">:</span> <span class="n">List</span><span class="p">[</span><span class="nb">int</span><span class="p">],</span>
@@ -246,7 +246,7 @@ <h1>Source code for neuralop.data.datasets.darcy</h1><div class="highlight"><pre
 <span class="n">example_data_root</span> <span class="o">=</span> <span class="n">get_project_root</span><span class="p">()</span> <span class="o">/</span> <span class="s2">&quot;neuralop/data/datasets/data&quot;</span>
 <div class="viewcode-block" id="load_darcy_flow_small">
 <a class="viewcode-back" href="../../../../modules/generated/neuralop.data.datasets.load_darcy_flow_small.html#neuralop.data.datasets.darcy.load_darcy_flow_small">[docs]</a>
-<span class="k">def</span> <span class="nf">load_darcy_flow_small</span><span class="p">(</span><span class="n">n_train</span><span class="p">,</span>
+<span class="k">def</span><span class="w"> </span><span class="nf">load_darcy_flow_small</span><span class="p">(</span><span class="n">n_train</span><span class="p">,</span>
     <span class="n">n_tests</span><span class="p">,</span>
     <span class="n">batch_size</span><span class="p">,</span>
     <span class="n">test_batch_sizes</span><span class="p">,</span>
@@ -290,7 +290,7 @@ <h1>Source code for neuralop.data.datasets.darcy</h1><div class="highlight"><pre
 
 
 <span class="c1"># legacy pt Darcy Flow loader</span>
-<span class="k">def</span> <span class="nf">load_darcy_pt</span><span class="p">(</span><span class="n">n_train</span><span class="p">,</span>
+<span class="k">def</span><span class="w"> </span><span class="nf">load_darcy_pt</span><span class="p">(</span><span class="n">n_train</span><span class="p">,</span>
                   <span class="n">n_tests</span><span class="p">,</span>
                   <span class="n">batch_size</span><span class="p">,</span>
                   <span class="n">test_batch_sizes</span><span class="p">,</span>