minor bug fix

Chilipp · Chilipp · commit 63d8502d8314 · 2018-03-24T11:17:44.000+01:00
diff --git a/psyplot/data.py b/psyplot/data.py
@@ -2147,8 +2147,13 @@ def base(self):
         if self._base is None:
             if 'variable' in self.arr.dims:
                 def to_dataset(i):
-                    return self.isel(variable=i).drop('variable').to_dataset(
+                    ret = self.isel(variable=i).to_dataset(
                         name=self.arr.coords['variable'].values[i])
+                    try:
+                        return ret.drop('variable')
+                    except ValueError:  # 'variable' Variable not defined
+                        pass
+                    return ret
                 ds = to_dataset(0)
                 if len(self.arr.coords['variable']) > 1:
                     for i in range(1, len(self.arr.coords['variable'])):
