Used black to try to avoid errors.

Author: gwgundersen

xarray/core/dataset.py

@@ -3452,8 +3452,7 @@ def _assert_all_in_dataset(
     # Drop variables
     @overload
     def drop(
-        self, labels: Union[Hashable, Iterable[Hashable]], *,
-            errors: str = "raise"
+        self, labels: Union[Hashable, Iterable[Hashable]], *, errors: str = "raise"
     ) -> "Dataset":
         ...
 
@@ -3464,7 +3463,7 @@ def drop(
     ) -> "Dataset":
         ...
 
-    def drop(self, labels=None, dim=None, *, errors='raise', **labels_kwargs):
+    def drop(self, labels=None, dim=None, *, errors="raise", **labels_kwargs):
         """Drop variables or index labels from this dataset.
 
         Parameters
@@ -3509,15 +3508,13 @@ def drop(self, labels=None, dim=None, *, errors='raise', **labels_kwargs):
         Data variables:
             A        (x, y) float64 -0.3944 -1.418 1.423 -1.041
         """
-        if errors not in ['raise', 'ignore']:
+        if errors not in ["raise", "ignore"]:
             raise ValueError('errors must be either "raise" or "ignore"')
 
         if labels_kwargs or utils.is_dict_like(labels):
-            labels_kwargs = utils.either_dict_or_kwargs(labels, labels_kwargs,
-                                                        'drop')
+            labels_kwargs = utils.either_dict_or_kwargs(labels, labels_kwargs, "drop")
             if dim is not None:
-                raise ValueError('cannot specify dim amd dict-like '
-                                 'arguments.')
+                raise ValueError("cannot specify dim amd dict-like " "arguments.")
             ds = self
             for dim, labels in labels_kwargs.items():
                 ds = ds._drop_labels(labels, dim, errors=errors)
@@ -3547,8 +3544,7 @@ def _drop_labels(self, labels=None, dim=None, errors="raise"):
         try:
             index = self.indexes[dim]
         except KeyError:
-            raise ValueError(
-                "dimension %r does not have coordinate labels" % dim)
+            raise ValueError("dimension %r does not have coordinate labels" % dim)
         new_index = index.drop(labels, errors=errors)
         return self.loc[{dim: new_index}]
 

xarray/tests/test_dataset.py

@@ -2163,29 +2163,28 @@ def test_drop_index_labels(self):
 
     def test_drop_labels_by_keyword(self):
         # Tests for #2910: Support for a additional `drop()` API.
-        data = Dataset({'A': (['x', 'y'], np.random.randn(2, 3)),
-                        'x': ['a', 'b']})
+        data = Dataset({"A": (["x", "y"], np.random.randn(2, 3)), "x": ["a", "b"]})
         # Basic functionality.
-        assert(len(data.coords['x']) == 2)
+        assert len(data.coords["x"]) == 2
 
         with pytest.warns(DeprecationWarning):
-            ds1 = data.drop(['a'], dim='x')
+            ds1 = data.drop(["a"], dim="x")
 
-        ds2 = data.drop(x='a')
-        ds3 = data.drop(x=['a'])
-        ds4 = data.drop(x=['a', 'b'])
-        assert(len(ds1.coords['x']) == 1)
-        assert(len(ds2.coords['x']) == 1)
-        assert(len(ds3.coords['x']) == 1)
-        assert(len(ds4.coords['x']) == 0)
+        ds2 = data.drop(x="a")
+        ds3 = data.drop(x=["a"])
+        ds4 = data.drop(x=["a", "b"])
+        assert len(ds1.coords["x"]) == 1
+        assert len(ds2.coords["x"]) == 1
+        assert len(ds3.coords["x"]) == 1
+        assert len(ds4.coords["x"]) == 0
 
         # Error handling if user tries both approaches.
         with pytest.raises(ValueError):
-            data.drop(labels=['a'], x='a')
+            data.drop(labels=["a"], x="a")
         with pytest.raises(ValueError):
-            data.drop(dim='x', x='a')
+            data.drop(dim="x", x="a")
         with pytest.raises(ValueError):
-            data.drop(labels=['a'], dim='x', x='a')
+            data.drop(labels=["a"], dim="x", x="a")
 
     def test_drop_dims(self):
         data = xr.Dataset(