Fix failing tests after numpy update (#120)

fmaussion · Aug 15, 2018 · 4d04d16 · 4d04d16
1 parent ce6482b
commit 4d04d16
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/salem/datasets.py b/salem/datasets.py
@@ -570,8 +570,8 @@ def __init__(self, x, y, crs=wgs84, size_x=640, size_y=640, scale=1,
         kwargs : **
           any keyword accepted by motionless.CenterMap (e.g. `key` for the API)
 
-        Note
-        ----
+        Notes
+        -----
         To obtain the exact domain specified in `x` and `y` you may have to
         play with the `size_x` and `size_y` kwargs.
         """

diff --git a/salem/wrftools.py b/salem/wrftools.py
@@ -429,7 +429,7 @@ def __getitem__(self, item):
         p = vars['P'][item] + vars['PB'][item]
         q = vars['QVAPOR'][item]
         z = (vars['PH'][item] + vars['PHB'][item]) / 9.81
-        return np.squeeze(_ncl_slp(z, tk, p, q), axis=squeezax)
+        return np.squeeze(_ncl_slp(z, tk, p, q), axis=tuple(squeezax))
 
 # Diagnostic variable classes in a list
 var_classes = [cls.__name__ for cls in vars()['FakeVariable'].__subclasses__()]