Fixing projection

Author: Chad-Peterson

examples/MVP/workflow.py

@@ -476,14 +476,14 @@
 total_time = (t3 - t1) / 1e9
 print(f"Total time: {total_time} seconds")
 
-# pf = prob.check_partials(includes='FEA')
+pf = prob.check_partials(includes='FEA')
 # tot = prob.compute_totals(of=['FEA.max_temperature'], wrt=['system.components.comp_2.translation'])
 # tot = tot[('FEA.max_temperature', 'system.components.comp_2.translation')][0]
 # print("total", tot)
-d_inputs = {'density': jnp.ones_like(densities_combined), 'heat_loads': jnp.ones_like(densities_combined)}
-d_outputs = {'temperature': jnp.ones_like(T), 'max_temperature': jnp.array(1)}
+# d_inputs = {'density': jnp.ones_like(densities_combined), 'heat_loads': jnp.ones_like(densities_combined)}
+# d_outputs = {'temperature': jnp.ones_like(T), 'max_temperature': jnp.array(1)}
 
-jvp_vals = prob.model.FEA.compute_jacvec_prod(prob.model.FEA._inputs, d_inputs, d_outputs, mode='fwd')
+# jvp_vals = prob.model.FEA.compute_jacvec_prod(prob.model.FEA._inputs, d_inputs, d_outputs, mode='fwd')
 # vjp_vals = prob.model.FEA.compute_jacvec_prod(prob.model.FEA._inputs, d_inputs, d_outputs, mode='rev')
 
 

src/SPI2py/API/FEA.py

@@ -176,7 +176,7 @@ def fea_solve(rhs):
 
         return u, u_max
 
-    def compute_jacvec_prod(self, inputs, d_inputs, d_outputs, mode):
+    def compute_jacvec_product(self, inputs, d_inputs, d_outputs, mode):
 
         # Unpack dynamic (differentiable) inputs.
         density = jnp.array(inputs["density"])
@@ -206,6 +206,7 @@ def compute_jacvec_prod(self, inputs, d_inputs, d_outputs, mode):
             d_T=dirichlet_values
         )
 
+
         if mode == "fwd":
             # Build tangents only for differentiable inputs.
             tan_density = (jnp.array(d_inputs["density"])