In save_probabilities(), drop selected dimension instead of all dims

2 jobs for dev-saving-missing_values
in 3 minutes and 27 seconds, using 0 compute credits, and was queued for 1 second