From da22607f7cc6fddc95263bf048c64bd42213eb21 Mon Sep 17 00:00:00 2001
From: Sathvik Bhagavan <sathvik.bhagavan@juliahub.com>
Date: Thu, 21 Mar 2024 08:36:15 +0000
Subject: [PATCH] test: remove tests for assertion error with batch to be true
 for QuadratureTraining

---
 test/NNODE_tests.jl | 7 -------
 1 file changed, 7 deletions(-)

diff --git a/test/NNODE_tests.jl b/test/NNODE_tests.jl
index 122adcceb3..b3731f059d 100644
--- a/test/NNODE_tests.jl
+++ b/test/NNODE_tests.jl
@@ -69,9 +69,6 @@ end
     sol = solve(prob, NNODE(luxchain, opt), verbose = true, maxiters = 400)
     @test sol.errors[:l2] < 0.5
 
-    @test_throws AssertionError solve(prob, NNODE(luxchain, opt; batch = true), verbose = true,
-                        maxiters = 400)
-
     sol = solve(prob,
                 NNODE(luxchain, opt; batch = false,
                                 strategy = StochasticTraining(100)),
@@ -105,10 +102,6 @@ end
                 abstol = 1.0f-8)
     @test sol.errors[:l2] < 0.5
 
-    @test_throws AssertionError solve(prob, NNODE(luxchain, opt; batch = true), verbose = true,
-                        maxiters = 400,
-                        abstol = 1.0f-8)
-
     sol = solve(prob,
                 NNODE(luxchain, opt; batch = false,
                                 strategy = StochasticTraining(100)),