def evaluator(make_ee_config): ensemble = TestEnsemble(0, 2, 1, 2) ee = EnsembleEvaluator( ensemble, make_ee_config(), 0, ee_id="ee-0", ) yield ee ee.stop()
def startSimulations(self, arg): with open(arg["config_file"], "r") as f: config = yaml.safe_load(f) ensemble = PrefectEnsemble(config) ee_config = arg["ee_config"] ee_id = str(uuid.uuid1()).split("-")[0] ee = EnsembleEvaluator(ensemble, ee_config, ee_id=ee_id) ee.run_and_get_successful_realizations() self._stop_time = time.time() self._queue_running = False ee.stop()
def evaluator(ee_config): ensemble = (create_ensemble_builder().add_realization( real=create_realization_builder().active(True).set_iens(0).add_stage( stage=create_stage_builder().add_step( step=create_step_builder().set_id(0).add_job( job=create_legacy_job_builder().set_id(0).set_name( "cat").set_ext_job(Mock())).add_job( job=create_legacy_job_builder().set_id(1).set_name( "cat2").set_ext_job(Mock())).set_dummy_io()). set_id(0).set_status("Unknown"))).set_ensemble_size(2).build()) ee = EnsembleEvaluator(ensemble=ensemble, config=ee_config) yield ee ee.stop()
def test_verify_dispatch_failing_job(make_ee_config): ee_config = make_ee_config(use_token=False, generate_cert=False) mock_ensemble = MagicMock() mock_ensemble.snapshot.to_dict.return_value = {} ee = EnsembleEvaluator( mock_ensemble, ee_config, 0, ee_id="0", ) ee.run() dispatch_failing_job.verify(ee_config.client_uri, on_connect=lambda: None) ee.stop()
def evaluator(ee_config): ensemble = (create_ensemble_builder().add_realization( real=create_realization_builder().active(True).set_iens(0).add_step( step=create_step_builder().set_id("0").set_name("cats").add_job( job=create_legacy_job_builder().set_id(0).set_name("cat"). set_ext_job(Mock())).add_job(job=create_legacy_job_builder( ).set_id(1).set_name("cat2").set_ext_job( Mock())).set_dummy_io())).set_ensemble_size(2).build()) ee = EnsembleEvaluator( ensemble, ee_config, 0, ee_id="ee-0", ) yield ee ee.stop()