Merge pull request #158 from tetrapod54/patch-1

[short_path] Define num_nodes in compute_to_go

Author: jstac

lectures/short_path.md

@@ -428,6 +428,7 @@ def bellman(J, Q):
 
 
 def compute_cost_to_go(Q):
+    num_nodes = Q.shape[0]
     J = np.zeros(num_nodes)      # Initial guess
     max_iter = 500
     i = 0