From 0011d6671dcc755930e44c56794bf963dc509aa6 Mon Sep 17 00:00:00 2001
From: Zachary Sunberg <sunbergzach@gmail.com>
Date: Sat, 1 Jun 2019 12:12:20 -0700
Subject: [PATCH] fix error with observation function

---
 src/discrete_explicit.jl | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/discrete_explicit.jl b/src/discrete_explicit.jl
index b21a59b..6b2d93a 100644
--- a/src/discrete_explicit.jl
+++ b/src/discrete_explicit.jl
@@ -54,6 +54,7 @@ POMDPModelTools.ordered_observations(m::DEP) = m.o
 # TODO support O(s, a, sp, o)
 # TODO initial state distribution
 # TODO convert_s, etc, dimensions
+# TODO better errors if T or Z return something unexpected
 
 """
     DiscreteExplicitPOMDP(S,A,O,T,Z,R,γ)
@@ -102,7 +103,7 @@ function DiscreteExplicitPOMDP(s, a, o, t, z, r, discount)
     m = DiscreteExplicitPOMDP(
         ss, as, os,
         tds, ods,
-        o, r,
+        z, r,
         Dict(ss[i]=>i for i in 1:length(ss)),
         Dict(as[i]=>i for i in 1:length(as)),
         Dict(os[i]=>i for i in 1:length(os)),