Update wet_snow.py

SnowEx · Sep 8, 2023 · 52a7c85 · 52a7c85
1 parent c9c3bb6
commit 52a7c85
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/spicy_snow/processing/wet_snow.py b/spicy_snow/processing/wet_snow.py
@@ -239,6 +239,6 @@ def flag_wet_snow(dataset: xr.Dataset, inplace: bool = False) -> Union[None, xr.
     # if less than 50% are wet then keep the save value for wet_snow otherwise set to 1
     dataset['wet_snow'] = dataset['wet_snow'].where(dataset['perma_wet'] < 0.5, 1)
 
-    # dataset['wet_snow'].loc[dict(time = ts)] = dataset.sel(time = ts)['wet_snow'].where(dataset.sel(time = ts)['ims'] == 4, 0)
+    dataset['wet_snow'].loc[dict(time = ts)] = dataset.sel(time = ts)['wet_snow'].where(dataset.sel(time = ts)['ims'] == 4, 0)
 
     return dataset