support for vector inputs

sisl · Mar 10, 2024 · ca07e21 · ca07e21
1 parent e71e4aa
commit ca07e21
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/epca.jl b/src/epca.jl
@@ -100,6 +100,7 @@ function CompressedBeliefMDPs.compress(epca::EPCA, X; verbose=false, maxiter::In
     end
     return Â
 end
-
+CompressedBeliefMDPs.compress(epca::EPCA, X::Vector; verbose=false, maxiter::Integer=50) = vec(compress(epca, X'; verbose=verbose, maxiter=maxiter))
 
 CompressedBeliefMDPs.decompress(epca::EPCA, A) = epca.g(A * epca.V)
+CompressedBeliefMDPs.decompress(epca::EPCA, A::Vector) = vec(epca.g(A' * epca.V))