Fix pytest

Author: tnipen

climetlab_maelstrom_yr/yr.py

@@ -187,9 +187,9 @@ def preprocess(self, ds):
             coords[coord] = ds.variables[coord]
         coords["predictor"] = np.concatenate((ds["predictor"].values, ds["static_predictor"].values))
         if self.probabilistic_target:
-            coords["target"] = np.array(["mean", "std"], np.object)
+            coords["target"] = np.array(["mean", "std"], object)
         else:
-            coords["target"] = np.array(["mean"], np.object)
+            coords["target"] = np.array(["mean"], object)
 
         # Copy variables from input
         data_vars = dict()

tests/test_yr.py

@@ -2,15 +2,17 @@
 
 import climetlab as cml
 import pandas as pd
+import os
 
 from climetlab_maelstrom_yr.yr import Yr
 
 def test_read():
+    dir = os.path.join(os.path.dirname(__file__), "files")
     ds = cml.load_dataset(
         "maelstrom-yr",
         size="5GB",
         dates=["2020-03-01", "2020-03-02"],
-        location="files/",
+        location=f"{dir}/",
         probabilistic_target=False,
         verbose=True,
     )