Update src/transformations.jl

Co-Authored-By: Milan Bouchet-Valat <[email protected]> Update src/transformations.jl Co-Authored-By: Milan Bouchet-Valat <[email protected]> Update src/transformations.jl Co-Authored-By: Milan Bouchet-Valat <[email protected]> Update src/transformations.jl Co-Authored-By: Milan Bouchet-Valat <[email protected]> Refactor
JuliaStats · Jul 8, 2019 · 434c556 · 434c556
1 parent 9fe549c
commit 434c556
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/src/transformations.jl b/src/transformations.jl
@@ -19,7 +19,7 @@ end
 
 function transform!(t::AbstractDataTransform, x::AbstractVector{<:Real})
     transform!(t, reshape(x, :, 1))
-    return vec(x)
+    return x
 end
 
 """
@@ -36,8 +36,7 @@ function transform(t::AbstractDataTransform, x::AbstractMatrix{<:Real})
 end
 
 function transform(t::AbstractDataTransform, x::AbstractVector{<:Real})
-    xmat = transform(t, reshape(x, :, 1))
-    return vec(xmat)
+    return vec(transform(t, reshape(x, :, 1)))
 end
 
 # reconstruct the original data from transformed values
@@ -58,7 +57,7 @@ end
 
 function reconstruct!(t::AbstractDataTransform, y::AbstractVector{<:Real})
     reconstruct!(t, reshape(y, :, 1))
-    return vec(y)
+    return y
 end
 
 """
@@ -76,8 +75,7 @@ function reconstruct(t::AbstractDataTransform, y::AbstractMatrix{<:Real})
 end
 
 function reconstruct(t::AbstractDataTransform, y::AbstractVector{<:Real})
-    ymat = reconstruct(t, reshape(y, :, 1))
-    return vec(ymat)
+    return vec(reconstruct(t, reshape(y, :, 1)))
 end
 
 """