Skip to content

Commit

Permalink
Remove ensemble generation from tests
Browse files Browse the repository at this point in the history
  • Loading branch information
al-rigazzi committed Aug 25, 2023
1 parent c4bf40c commit d350dd0
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 10 deletions.
6 changes: 1 addition & 5 deletions tests/backends/test_dbmodel.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down Expand Up @@ -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)
Expand Down Expand Up @@ -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"
)
)
6 changes: 1 addition & 5 deletions tests/backends/test_dbscript.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand All @@ -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()
Expand Down Expand Up @@ -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)
Expand Down

0 comments on commit d350dd0

Please sign in to comment.