Commit 80a0b065 authored by Steven Cordwell's avatar Steven Cordwell
Browse files

matrix() is required

parent c735b5f2
......@@ -717,7 +717,7 @@ class MDP(object):
except AttributeError:
raise TypeError("bellman: V must be a numpy array or matrix.")
Q = empty((self.S, self.A))
Q = matrix(empty((self.S, self.A)))
for aa in range(self.A):
Q[:, aa] = self.R[:, aa] + (self.discount * self.P[aa] * V)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment