Fix a bug on the dimension of toy exampe (#448)

StanfordASL · Nov 1, 2023 · 5ede47b · 5ede47b
1 parent 32a44d4
commit 5ede47b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neural_network_lyapunov/test/train_toy_system_controller_demo.py b/neural_network_lyapunov/test/train_toy_system_controller_demo.py
@@ -82,7 +82,7 @@
     dut = train_lyapunov_barrier.Trainer()
     dut.add_lyapunov(
         lyapunov_hybrid_system, V_lambda, x_equilibrium,
-        r_options.FixedROptions(torch.eye(2, dtype=torch.float64)))
+        r_options.FixedROptions(torch.eye(args.dimension, dtype=torch.float64)))
     dut.lyapunov_positivity_mip_pool_solutions = 1
     dut.lyapunov_derivative_mip_pool_solutions = 1
     dut.lyapunov_derivative_convergence_tol = 5E-5