From d350dd08022ec6dd042708cbc87836556a9a3644 Mon Sep 17 00:00:00 2001 From: Ale Rigazzi Date: Fri, 25 Aug 2023 10:25:42 -0500 Subject: [PATCH] Remove ensemble generation from tests --- tests/backends/test_dbmodel.py | 6 +----- tests/backends/test_dbscript.py | 6 +----- 2 files changed, 2 insertions(+), 10 deletions(-) diff --git a/tests/backends/test_dbmodel.py b/tests/backends/test_dbmodel.py index 96c5d54e9..83a2e119b 100644 --- a/tests/backends/test_dbmodel.py +++ b/tests/backends/test_dbmodel.py @@ -585,8 +585,6 @@ def test_colocated_db_model_ensemble(fileutils, wlmutils, mlutils): outputs=outputs2, ) - exp.generate(colo_ensemble, overwrite=True) - # Launch and check successful completion try: exp.start(colo_ensemble, block=True) @@ -688,8 +686,6 @@ def test_colocated_db_model_ensemble_reordered(fileutils, wlmutils, mlutils): outputs=outputs2, ) - exp.generate(colo_ensemble, overwrite=True) - # Launch and check successful completion try: exp.start(colo_ensemble, block=True) @@ -816,4 +812,4 @@ def test_inconsistent_params_db_model(): assert ( ex.value.args[0] == "Cannot set devices_per_node>1 if CPU is specified under devices" - ) \ No newline at end of file + ) diff --git a/tests/backends/test_dbscript.py b/tests/backends/test_dbscript.py index ef5b786e4..06492f60f 100644 --- a/tests/backends/test_dbscript.py +++ b/tests/backends/test_dbscript.py @@ -369,8 +369,6 @@ def test_colocated_db_script_ensemble(fileutils, wlmutils, mlutils): # Assert we have added both models to each entity assert all([len(entity._db_scripts) == 2 for entity in colo_ensemble]) - exp.generate(colo_ensemble, overwrite=True) - # Launch and check successful completion try: exp.start(colo_ensemble, block=True) @@ -386,7 +384,7 @@ def test_colocated_db_script_ensemble_reordered(fileutils, wlmutils, mlutils): script to the ensemble, then colocating the DB""" # Set Experiment name - exp_name = "test-colocated-db-script" + exp_name = "test-colocated-db-script-reord" # Retrieve parameters from testing environment test_launcher = wlmutils.get_test_launcher() @@ -466,8 +464,6 @@ def test_colocated_db_script_ensemble_reordered(fileutils, wlmutils, mlutils): # Assert we have added both models to each entity assert all([len(entity._db_scripts) == 2 for entity in colo_ensemble]) - exp.generate(colo_ensemble, overwrite=True) - # Launch and check successful completion try: exp.start(colo_ensemble, block=True)