ERROR: TypeError: fitdbm: in typeassert, expected Array{Int64,1}, got Array{Int64,2}

in trainpartdbm

function trainpartdbm(mat;partitions::Array{Array{Int64,1},1}=[collect(1:size(mat)[2])],

nhiddens::Array{Int64,1}=[size(mat)[2],div(size(mat)[2],10)],

epochs::Int64=20,

nparticles::Int64=100,

learningrate::Float64=0.005)

```
partDBMS = Vector{BoltzmannMachines.BasicDBM}()
nhiddensmat = partitionhiddens(partitions,nhiddens)
println(nhiddensmat)
for i=1:length(partitions)
partx = mat[:,partitions[i]]
pdbm = BoltzmannMachines.fitdbm(partx,
nhiddens=nhiddensmat[i,:],
epochs =epochs,
nparticles=nparticles,
learningrate=learningrate
)
push!(partDBMS,pdbm)
end
jointdbm = BoltzmannMachines.joindbms(partDBMS,partitions)
jointdbm,partDBMS
```

end

end

Kinsly suggest me how to solve it. Thanks in advance