Skip to content

Commit

Permalink
distributed mixed_poisson in work
Browse files Browse the repository at this point in the history
  • Loading branch information
niravshah241 committed Jul 19, 2024
1 parent 92ee30e commit 813498a
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -471,7 +471,7 @@ def generate_ann_output_set_sigma(problem, reduced_problem, input_set,
itemsize = MPI.DOUBLE.Get_size()

if world_comm.rank == 0:
ann_input_set_sigma = generate_ann_input_set(samples=num_ann_samples_sigma)
ann_input_set_sigma = generate_ann_input_set(num_ann_samples=num_ann_samples_sigma)
np.random.shuffle(ann_input_set_sigma)
nbytes_para_ann_training_sigma = num_training_samples_sigma * itemsize * para_dim_sigma
nbytes_dofs_ann_training_sigma = num_training_samples_sigma * itemsize * \
Expand Down

0 comments on commit 813498a

Please sign in to comment.