Merge pull request #29 from invenia/wct/make-indep-mogp-fast

collect -> Matrix
JuliaGaussianProcesses · Aug 25, 2021 · d005a76 · d005a76 · willtebbutt · Aug 25, 2021
2 parents cdc4cf7 + d51107b
commit d005a76
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "LinearMixingModels"
 uuid = "b8ce4b42-e81b-4a39-a84a-67f74a9a16dd"
 authors = ["Invenia Technical Computing Corporation"]
-version = "0.1.2"
+version = "0.1.3"
 
 [deps]
 AbstractGPs = "99985d1d-32ba-4be9-9821-2ec096f28918"

diff --git a/src/independent_mogp.jl b/src/independent_mogp.jl
@@ -61,17 +61,15 @@ end
 # See AbstractGPs.jl API docs.
 function AbstractGPs.cov(f::IndependentMOGP, x::MOInputIsotopicByOutputs)
     Cs = map(f -> cov(f, x.x), f.fs)
-    return collect(BlockDiagonal(Cs))
+    return Matrix(BlockDiagonal(Cs))
 end
 
 # See AbstractGPs.jl API docs.
 function AbstractGPs.cov(
-    f::IndependentMOGP,
-    x::MOInputIsotopicByOutputs,
-    y::MOInputIsotopicByOutputs,
+    f::IndependentMOGP, x::MOInputIsotopicByOutputs, y::MOInputIsotopicByOutputs
 )
     Cs = map(f -> cov(f, x.x, y.x), f.fs)
-    return collect(BlockDiagonal(Cs))
+    return Matrix(BlockDiagonal(Cs))
 end
 
 # See AbstractGPs.jl API docs.