From 757e8cfc286485b95577fb51ccbef9cf57c1814e Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" <41898282+github-actions[bot]@users.noreply.github.com> Date: Thu, 20 Aug 2020 18:10:59 +0000 Subject: [PATCH] Format .jl files --- src/components/explorers/batch_explorer.jl | 5 +++-- src/components/explorers/gumbel_softmax_explorer.jl | 5 ++++- src/components/explorers/weighted_softmax_explorer.jl | 5 ++--- 3 files changed, 9 insertions(+), 6 deletions(-) diff --git a/src/components/explorers/batch_explorer.jl b/src/components/explorers/batch_explorer.jl index 45cf384..f81bd7d 100644 --- a/src/components/explorers/batch_explorer.jl +++ b/src/components/explorers/batch_explorer.jl @@ -16,9 +16,10 @@ Apply inner explorer to each column of `values`. """ (x::BatchExplorer)(values::AbstractMatrix) = [x.explorer(v) for v in eachcol(values)] -(x::BatchExplorer)(values::AbstractMatrix, mask::AbstractMatrix) = [x.explorer(v,m) for (v,m) in zip(eachcol(values), eachcol(mask))] +(x::BatchExplorer)(values::AbstractMatrix, mask::AbstractMatrix) = + [x.explorer(v, m) for (v, m) in zip(eachcol(values), eachcol(mask))] (x::BatchExplorer)(v::AbstractVector) = x.explorer(v) -(x::BatchExplorer)(v::AbstractVector, m::AbstractVector) = x.explorer(v,m) +(x::BatchExplorer)(v::AbstractVector, m::AbstractVector) = x.explorer(v, m) Flux.testmode!(x::BatchExplorer, mode = true) = testmode!(x.explorer, mode) diff --git a/src/components/explorers/gumbel_softmax_explorer.jl b/src/components/explorers/gumbel_softmax_explorer.jl index 5e2861e..ba8f29b 100644 --- a/src/components/explorers/gumbel_softmax_explorer.jl +++ b/src/components/explorers/gumbel_softmax_explorer.jl @@ -15,7 +15,10 @@ function (p::GumbelSoftmaxExplorer)(v::AbstractVector{T}) where {T} argmax(logits .- log.(-log.(u))) end -function (p::GumbelSoftmaxExplorer)(v::AbstractVector{T}, mask::AbstractVector{Bool}) where {T} +function (p::GumbelSoftmaxExplorer)( + v::AbstractVector{T}, + mask::AbstractVector{Bool}, +) where {T} v[.!mask] .= typemin(T) p(v) end diff --git a/src/components/explorers/weighted_softmax_explorer.jl b/src/components/explorers/weighted_softmax_explorer.jl index 0bdf5d3..12747ca 100644 --- a/src/components/explorers/weighted_softmax_explorer.jl +++ b/src/components/explorers/weighted_softmax_explorer.jl @@ -13,15 +13,14 @@ struct WeightedSoftmaxExplorer{R<:AbstractRNG} <: AbstractExplorer rng::R end -function WeightedSoftmaxExplorer(;rng = Random.GLOBAL_RNG) +function WeightedSoftmaxExplorer(; rng = Random.GLOBAL_RNG) WeightedSoftmaxExplorer(rng) end (s::WeightedSoftmaxExplorer)(values::AbstractVector{T}) where {T} = sample(s.rng, Weights(softmax(values), one(T))) -function (s::WeightedSoftmaxExplorer)(values::AbstractVector{T}, mask) where T +function (s::WeightedSoftmaxExplorer)(values::AbstractVector{T}, mask) where {T} values[.!mask] .= typemin(T) s(values) end -