Skip to content

Commit 0a309e0

Browse files
authored
[skip-ci] Fix some asv benchmarks (#4920)
Mostly adjust to deprecated behaviour
1 parent 9858a27 commit 0a309e0

File tree

3 files changed

+5
-11
lines changed

3 files changed

+5
-11
lines changed

asv_bench/benchmarks/combine.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -26,13 +26,13 @@ def setup(self):
2626
{"B": xr.DataArray(data, coords={"T": t + t_size}, dims=("T", "X", "Y"))}
2727
)
2828

29-
def time_combine_manual(self):
29+
def time_combine_nested(self):
3030
datasets = [[self.dsA0, self.dsA1], [self.dsB0, self.dsB1]]
3131

32-
xr.combine_manual(datasets, concat_dim=[None, "t"])
32+
xr.combine_nested(datasets, concat_dim=[None, "T"])
3333

34-
def time_auto_combine(self):
34+
def time_combine_by_coords(self):
3535
"""Also has to load and arrange t coordinate"""
3636
datasets = [self.dsA0, self.dsA1, self.dsB0, self.dsB1]
3737

38-
xr.combine_auto(datasets)
38+
xr.combine_by_coords(datasets)

asv_bench/benchmarks/dataset_io.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -59,23 +59,20 @@ def make_ds(self):
5959
coords={"lon": lons, "lat": lats, "time": times},
6060
dims=("time", "lon", "lat"),
6161
name="foo",
62-
encoding=None,
6362
attrs={"units": "foo units", "description": "a description"},
6463
)
6564
self.ds["bar"] = xr.DataArray(
6665
randn((self.nt, self.nx, self.ny), frac_nan=0.2),
6766
coords={"lon": lons, "lat": lats, "time": times},
6867
dims=("time", "lon", "lat"),
6968
name="bar",
70-
encoding=None,
7169
attrs={"units": "bar units", "description": "a description"},
7270
)
7371
self.ds["baz"] = xr.DataArray(
7472
randn((self.nx, self.ny), frac_nan=0.2).astype(np.float32),
7573
coords={"lon": lons, "lat": lats},
7674
dims=("lon", "lat"),
7775
name="baz",
78-
encoding=None,
7976
attrs={"units": "baz units", "description": "a description"},
8077
)
8178

@@ -270,23 +267,20 @@ def make_ds(self, nfiles=10):
270267
coords={"lon": lons, "lat": lats, "time": times},
271268
dims=("time", "lon", "lat"),
272269
name="foo",
273-
encoding=None,
274270
attrs={"units": "foo units", "description": "a description"},
275271
)
276272
ds["bar"] = xr.DataArray(
277273
randn((nt, self.nx, self.ny), frac_nan=0.2),
278274
coords={"lon": lons, "lat": lats, "time": times},
279275
dims=("time", "lon", "lat"),
280276
name="bar",
281-
encoding=None,
282277
attrs={"units": "bar units", "description": "a description"},
283278
)
284279
ds["baz"] = xr.DataArray(
285280
randn((self.nx, self.ny), frac_nan=0.2).astype(np.float32),
286281
coords={"lon": lons, "lat": lats},
287282
dims=("lon", "lat"),
288283
name="baz",
289-
encoding=None,
290284
attrs={"units": "baz units", "description": "a description"},
291285
)
292286

asv_bench/benchmarks/repr.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44

55

66
class ReprMultiIndex:
7-
def setup(self, key):
7+
def setup(self):
88
index = pd.MultiIndex.from_product(
99
[range(10000), range(10000)], names=("level_0", "level_1")
1010
)

0 commit comments

Comments
 (0)