fixes

MartinAchondo · Jul 2, 2024 · 91d3f63 · 91d3f63
1 parent 0c0706e
commit 91d3f63
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_xppbe.py b/tests/test_xppbe.py
@@ -59,10 +59,10 @@ def test_scripts():
 @pytest.mark.parametrize(
  ('pinns_method','model','scheme'),
  (
-     ('DCM','linear','direct'),
+     #('DCM','linear','direct'),
      #('DCM','linear', 'regularized_scheme_1'),
      #('DCM','nonlinear','regularized_scheme_2'),
-     ('DVM','linear','direct'),
+     #('DVM','linear','direct'),
      #('DVM','linear','regularized_scheme_1'),
      #('DVM','nonlinear','regularized_scheme_1'),
      ('DBM','linear','direct')
@@ -81,7 +81,7 @@ def test_pinns_method_and_schemes(pinns_method,model,scheme):
         sim.equation = scheme
         if pinns_method == 'DBM':
             sim.num_networks = 1
-            sim.losses['IB1','IB2']
+            sim.losses = ['IB1','IB2']
         sim.create_simulation()
         sim.adapt_model()
         sim.solve_model()