Mixed poisson distributed in work

Wells-Group · Jul 18, 2024 · 64f668a · 64f668a
1 parent aa5c581
commit 64f668a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/demo/mixed_poisson_dlrbnicsx_distributed/dlrbnicsx_mixed_poisson_distributed.py b/demo/mixed_poisson_dlrbnicsx_distributed/dlrbnicsx_mixed_poisson_distributed.py
@@ -368,7 +368,7 @@ def generate_training_set(samples=num_pod_samples_sigma):
 for mu_index in cpu_indices_sigma:
     print(f"Parameter number {mu_index+1} of {para_matrix_sigma.shape[0]}: {para_matrix_sigma[mu_index,:]}")
     solution_sigma, solution_u = problem_parametric.solve(para_matrix_sigma[mu_index, :])
-    snapshot_arrays_sigma[mu_index, rstart_sigma:rend_sigma] = solution.vector[rstart_sigma:rend_sigma]
+    snapshot_arrays_sigma[mu_index, rstart_sigma:rend_sigma] = solution_sigma.vector[rstart_sigma:rend_sigma]
 
 world_comm.barrier()