Skip to content

Commit d6b86cb

Browse files
committed
fixing bug in Distribution types
1 parent 1c4118b commit d6b86cb

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/flows/realnvp.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -153,7 +153,7 @@ end
153153
154154
Default constructor of RealNVP flow, which is a composition of `nlayers` RealNVP_layer.
155155
# Arguments
156-
- `q0::Distribution{Continuous, Multivariate}`: reference distribution, e.g. `MvNormal(zeros(dims), I)`
156+
- `q0::Distribution{Multivariate,Continuous}`: reference distribution, e.g. `MvNormal(zeros(dims), I)`
157157
- `dims::Int`: dimension of problem
158158
- `hdims::AbstractVector{Int}`: dimension of hidden units for s and t
159159
- `nlayers::Int`: number of RealNVP_layer
@@ -166,7 +166,7 @@ Default constructor of RealNVP flow, which is a composition of `nlayers` RealNVP
166166
"""
167167

168168
function realnvp(
169-
q0::Distribution{Continuous, Multivariate},
169+
q0::Distribution{Multivariate,Continuous},
170170
hdims::AbstractVector{Int}, # dimension of hidden units for s and t
171171
nlayers::Int; # number of RealNVP_layer
172172
paramtype::Type{T} = Float64, # type of the parameters

0 commit comments

Comments
 (0)