Skip to content

Commit 28fd363

Browse files
committed
rm kw combinechunks
1 parent ec75b56 commit 28fd363

File tree

2 files changed

+40
-40
lines changed

2 files changed

+40
-40
lines changed

src/DatasetAPI/Datasets.jl

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -348,7 +348,7 @@ open_mfdataset(g::AbstractString; kwargs...) = open_mfdataset(_glob(g); kwargs..
348348
open_mfdataset(g::Vector{<:AbstractString}; kwargs...) =
349349
merge_datasets(map(i -> open_dataset(i; kwargs...), g))
350350

351-
function merge_new_axis(alldatasets, firstcube,var,mergedim,combinechunks)
351+
function merge_new_axis(alldatasets, firstcube,var,mergedim)
352352
newdim = if !(typeof(DD.lookup(mergedim)) <: DD.NoLookup)
353353
DD.rebuild(mergedim, DD.val(mergedim))
354354
else
@@ -359,23 +359,23 @@ function merge_new_axis(alldatasets, firstcube,var,mergedim,combinechunks)
359359
end
360360
newdims = (DD.dims(firstcube)...,newdim)
361361
s = ntuple(i->i==length(newdims) ? length(alldiskarrays) : 1, length(newdims))
362-
newda = DiskArrays.ConcatDiskArray(reshape(alldiskarrays,s...),combinechunks)
362+
newda = DiskArrays.ConcatDiskArray(reshape(alldiskarrays,s...))
363363
YAXArray(newdims,newda,deepcopy(firstcube.properties))
364364
end
365-
function merge_existing_axis(alldatasets,firstcube,var,mergedim,combinechunks)
365+
function merge_existing_axis(alldatasets,firstcube,var,mergedim)
366366
allaxvals = map(ds->DD.dims(ds.cubes[var],mergedim).val,alldatasets)
367367
newaxvals = reduce(vcat,allaxvals)
368368
newdim = DD.rebuild(mergedim,newaxvals)
369369
alldiskarrays = map(ds->ds.cubes[var].data,alldatasets)
370370
istack = DD.dimnum(firstcube,mergedim)
371371
newshape = ntuple(i->i!=istack ? 1 : length(alldiskarrays),ndims(firstcube))
372-
newda = DiskArrays.ConcatDiskArray(reshape(alldiskarrays,newshape),combinechunks)
372+
newda = DiskArrays.ConcatDiskArray(reshape(alldiskarrays,newshape))
373373
newdims = Base.setindex(firstcube.axes,newdim,istack)
374374
YAXArray(newdims,newda,deepcopy(firstcube.properties))
375375
end
376376

377377
"""
378-
open_mfdataset(files::DD.DimVector{<:AbstractString}; combinechunks=:error, kwargs...)
378+
open_mfdataset(files::DD.DimVector{<:AbstractString}; kwargs...)
379379
380380
Opens and concatenates a list of dataset paths along the dimension specified in `files`.
381381
@@ -397,7 +397,7 @@ files = ["a.nc", "b.nc", "c.nc"]
397397
open_mfdataset(DD.DimArray(files, DD.Dim{:NewDim}(["a","b","c"])))
398398
````
399399
"""
400-
function open_mfdataset(vec::DD.DimVector{<:Union{Missing,AbstractString}};combinechunks=:maxsize, kwargs...)
400+
function open_mfdataset(vec::DD.DimVector{<:Union{Missing,AbstractString}}; kwargs...)
401401
alldatasets = map(vec) do filename
402402
ismissing(filename) ? missing : open_dataset(filename;kwargs...)
403403
end
@@ -407,9 +407,9 @@ function open_mfdataset(vec::DD.DimVector{<:Union{Missing,AbstractString}};combi
407407
ars = map(vars_to_merge) do var
408408
cfi = fi.cubes[var]
409409
mergedar = if DD.dims(cfi,mergedim) !== nothing
410-
merge_existing_axis(alldatasets,cfi,var,mergedim,combinechunks)
410+
merge_existing_axis(alldatasets,cfi,var,mergedim)
411411
else
412-
merge_new_axis(alldatasets,cfi,var,mergedim,combinechunks)
412+
merge_new_axis(alldatasets,cfi,var,mergedim)
413413
end
414414
var => mergedar
415415
end

test/Datasets/datasets.jl

Lines changed: 32 additions & 32 deletions
Original file line numberDiff line numberDiff line change
@@ -540,35 +540,35 @@ end
540540
@test_throws ArgumentError Cube(ds)
541541
end
542542

543-
# @testset "Open_mfdataset" begin
544-
# import NetCDF
545-
# d1 = DD.X(1:20)
546-
# d2 = DD.Y(1:10)
547-
# a1 = rand(20,10)
548-
# a2 = rand(20,10)
549-
# td = mktempdir()
550-
# f1, f2 = joinpath.(td,("file_1.nc","file_2.nc"))
551-
552-
# array1 = YAXArray((d1,d2),a1)
553-
# array2 = YAXArray((d1,d2),a2)
554-
555-
# savecube(array1,f1)
556-
# savecube(array2,f2)
557-
558-
# ds = open_mfdataset(DD.DimArray([f1,f2],(YAX.time(1:2),)))
559-
560-
# @test ds.layer.data[:,:,1] == array1
561-
# @test ds.layer.data[:,:,2] == array2
562-
563-
# td = mktempdir()
564-
# f1, f2 = joinpath.(td,("file_1.nc","file_2.nc"))
565-
# td1, td2 = YAX.time(1:2), YAX.time(3:4)
566-
# a1,a2 = rand(20,10,2), rand(20,10,2)
567-
# array1,array2 = YAXArray((d1,d2,td1),a1), YAXArray((d1,d2,td2),a2)
568-
# savecube(array1,f1)
569-
# savecube(array2,f2)
570-
571-
# ds = open_mfdataset(joinpath(td,"*.nc"))
572-
# @test ds.layer.data[:,:,1:2] == array1
573-
# @test ds.layer.data[:,:,3:4] == array2
574-
# end
543+
@testset "Open_mfdataset" begin
544+
import NetCDF
545+
d1 = DD.X(1:20)
546+
d2 = DD.Y(1:10)
547+
a1 = rand(20,10)
548+
a2 = rand(20,10)
549+
td = mktempdir()
550+
f1, f2 = joinpath.(td,("file_1.nc","file_2.nc"))
551+
552+
array1 = YAXArray((d1,d2),a1)
553+
array2 = YAXArray((d1,d2),a2)
554+
555+
savecube(array1,f1)
556+
savecube(array2,f2)
557+
558+
ds = open_mfdataset(DD.DimArray([f1,f2],(YAX.time(1:2),)))
559+
560+
@test ds.layer.data[:,:,1] == array1
561+
@test ds.layer.data[:,:,2] == array2
562+
563+
td = mktempdir()
564+
f1, f2 = joinpath.(td,("file_1.nc","file_2.nc"))
565+
td1, td2 = YAX.time(1:2), YAX.time(3:4)
566+
a1,a2 = rand(20,10,2), rand(20,10,2)
567+
array1,array2 = YAXArray((d1,d2,td1),a1), YAXArray((d1,d2,td2),a2)
568+
savecube(array1,f1)
569+
savecube(array2,f2)
570+
571+
ds = open_mfdataset(joinpath(td,"*.nc"))
572+
@test ds.layer.data[:,:,1:2] == array1
573+
@test ds.layer.data[:,:,3:4] == array2
574+
end

0 commit comments

Comments
 (0)