Python FEniCS/ufl@b7c3ad7

Author: FEniCS GitHub Actions

ufl/main/_modules/ufl/algorithms/apply_derivatives.html

@@ -46,7 +46,7 @@ <h1>Source code for ufl.algorithms.apply_derivatives</h1><div class="highlight">
 <span class="kn">import</span><span class="w"> </span><span class="nn">numpy</span><span class="w"> </span><span class="k">as</span><span class="w"> </span><span class="nn">np</span>
 
 <span class="kn">from</span><span class="w"> </span><span class="nn">ufl.action</span><span class="w"> </span><span class="kn">import</span> <span class="n">Action</span>
-<span class="kn">from</span><span class="w"> </span><span class="nn">ufl.algorithms.analysis</span><span class="w"> </span><span class="kn">import</span> <span class="n">extract_arguments</span>
+<span class="kn">from</span><span class="w"> </span><span class="nn">ufl.algorithms.analysis</span><span class="w"> </span><span class="kn">import</span> <span class="n">extract_arguments</span><span class="p">,</span> <span class="n">extract_coefficients</span>
 <span class="kn">from</span><span class="w"> </span><span class="nn">ufl.algorithms.map_integrands</span><span class="w"> </span><span class="kn">import</span> <span class="n">map_integrand_dags</span>
 <span class="kn">from</span><span class="w"> </span><span class="nn">ufl.algorithms.replace_derivative_nodes</span><span class="w"> </span><span class="kn">import</span> <span class="n">replace_derivative_nodes</span>
 <span class="kn">from</span><span class="w"> </span><span class="nn">ufl.argument</span><span class="w"> </span><span class="kn">import</span> <span class="n">BaseArgument</span>
@@ -1419,9 +1419,12 @@ <h1>Source code for ufl.algorithms.apply_derivatives</h1><div class="highlight">
 <a class="viewcode-back" href="../../../api-doc/ufl.algorithms.html#ufl.algorithms.apply_derivatives.GateauxDerivativeRuleset.reference_grad">[docs]</a>
     <span class="k">def</span><span class="w"> </span><span class="nf">reference_grad</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">o</span><span class="p">):</span>
 <span class="w">        </span><span class="sd">&quot;&quot;&quot;Differentiate a reference_grad.&quot;&quot;&quot;</span>
-        <span class="k">raise</span> <span class="ne">NotImplementedError</span><span class="p">(</span>
-            <span class="s2">&quot;Currently no support for ReferenceGrad in CoefficientDerivative.&quot;</span>
-        <span class="p">)</span></div>
+        <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="n">extract_coefficients</span><span class="p">(</span><span class="n">o</span><span class="p">))</span> <span class="o">&gt;</span> <span class="mi">0</span><span class="p">:</span>
+            <span class="k">raise</span> <span class="ne">NotImplementedError</span><span class="p">(</span>
+                <span class="s2">&quot;Currently no support for ReferenceGrad in CoefficientDerivative.&quot;</span>
+            <span class="p">)</span>
+        <span class="k">else</span><span class="p">:</span>
+            <span class="k">return</span> <span class="n">Zero</span><span class="p">(</span><span class="n">o</span><span class="o">.</span><span class="n">ufl_shape</span><span class="p">)</span></div>
 
         <span class="c1"># TODO: This is implementable for regular derivative(M(f),f,v)</span>
         <span class="c1">#       but too messy if customized coefficient derivative</span>