fix bug

igmhub · Sep 30, 2024 · f5ac5ef · f5ac5ef
1 parent 040778d
commit f5ac5ef
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/plot_mpg.py b/tests/plot_mpg.py
@@ -41,7 +41,7 @@ def test():
             emulator = NNEmulator(
                 training_set=training_set,
                 emulator_label=emulator_label,
-                emu_params=nyx_emu_params,
+                emu_params=nyx_params,
                 model_path=model_path,
                 drop_sim=None,
                 train=False,
@@ -50,7 +50,7 @@ def test():
             emulator = NNEmulator(
                 training_set=training_set,
                 emulator_label=emulator_label,
-                emu_params=nyx_emu_params,
+                emu_params=nyx_params,
                 model_path=model_path+f'_{sim}.pt',
                 drop_sim=sim,
                 train=False,