Skip to content

Commit

Permalink
Replace drop_sel with drop_var everywhere in __main__.py
Browse files Browse the repository at this point in the history
We received "ValueError: the first argument to .drop_sel must be a
dictionary" as mentioned in monocongo#495. Replacing .drop_sel with .drop_var
resolved the issue. A similar fix was proposed in monocongo#493, but some
calls to .drop_sel remained.
  • Loading branch information
dawiedotcom committed Mar 29, 2023
1 parent db85494 commit bb61f98
Showing 1 changed file with 7 additions and 7 deletions.
14 changes: 7 additions & 7 deletions src/climate_indices/__main__.py
Original file line number Diff line number Diff line change
Expand Up @@ -679,7 +679,7 @@ def _drop_data_into_shared_arrays_divisions(dataset,
output_shape = dataset[var_name].shape

# drop the variable from the dataset (we're assuming this frees the memory)
dataset = dataset.drop_sel(var_name)
dataset = dataset.drop_vars(names=[var_name])

return output_shape

Expand Down Expand Up @@ -744,7 +744,7 @@ def _compute_write_index(keyword_arguments):
input_var_names.append("lat")
for var in dataset.data_vars:
if var not in input_var_names:
dataset = dataset.drop_sel(var)
dataset = dataset.drop_vars(names=[var])

# get the initial year of the data
data_start_year = int(str(dataset["time"].values[0])[0:4])
Expand Down Expand Up @@ -920,7 +920,7 @@ def _compute_write_index(keyword_arguments):
# remove all data variables except for the new SCPDSI variable
for var_name in dataset.data_vars:
if var_name != var_name_scpdsi:
dataset = dataset.drop_sel(var_name)
dataset = dataset.drop_vars(names=[var_name])

# TODO set global attributes accordingly for this new dataset

Expand All @@ -941,7 +941,7 @@ def _compute_write_index(keyword_arguments):
# remove all data variables except for the new PDSI variable
for var_name in dataset.data_vars:
if var_name != var_name_pdsi:
dataset = dataset.drop_sel(var_name)
dataset = dataset.drop_vars(names=[var_name])

# TODO set global attributes accordingly for this new dataset

Expand All @@ -962,7 +962,7 @@ def _compute_write_index(keyword_arguments):
# remove all data variables except for the new PHDI variable
for var_name in dataset.data_vars:
if var_name != var_name_phdi:
dataset = dataset.drop_sel(var_name)
dataset = dataset.drop_vars(names=[var_name])

# TODO set global attributes accordingly for this new dataset

Expand All @@ -983,7 +983,7 @@ def _compute_write_index(keyword_arguments):
# remove all data variables except for the new PMDI variable
for var_name in dataset.data_vars:
if var_name != var_name_pmdi:
dataset = dataset.drop_sel(var_name)
dataset = dataset.drop_vars(names=[var_name])

# TODO set global attributes accordingly for this new dataset

Expand All @@ -1004,7 +1004,7 @@ def _compute_write_index(keyword_arguments):
# remove all data variables except for the new Z-Index variable
for var_name in dataset.data_vars:
if var_name != var_name_zindex:
dataset = dataset.drop_sel(var_name)
dataset = dataset.drop_vars(names=[var_name])

# TODO set global attributes accordingly for this new dataset

Expand Down

0 comments on commit bb61f98

Please sign in to comment.