Skip to content

Commit

Permalink
additional method logging
Browse files Browse the repository at this point in the history
  • Loading branch information
quaquel committed Mar 20, 2024
1 parent c26009c commit fc3fff4
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion ema_workbench/em_framework/futures_mpi.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@
from .util import NamedObjectMap
from .model import AbstractModel
from .experiment_runner import ExperimentRunner
from ..util import get_module_logger, get_rootlogger
from ..util import get_module_logger, get_rootlogger, method_logger

from ..util import ema_logging

Expand Down Expand Up @@ -154,6 +154,7 @@ def __init__(self, msis, n_processes=None, **kwargs):
self.stop_event = None
self.n_processes = n_processes

@method_logger(__name__)
def initialize(self):
# Only import mpi4py if the MPIEvaluator is used, to avoid unnecessary dependencies.
from mpi4py.futures import MPIPoolExecutor
Expand All @@ -178,6 +179,7 @@ def initialize(self):
)
return self

@method_logger(__name__)
def finalize(self):
self._pool.shutdown()
self.stop_event.set()
Expand All @@ -189,6 +191,7 @@ def finalize(self):
time.sleep(0.1)
_logger.info("MPI pool has been shut down")

@method_logger(__name__)
def evaluate_experiments(self, scenarios, policies, callback, combine="factorial", **kwargs):
ex_gen = experiment_generator(scenarios, self._msis, policies, combine=combine)
experiments = list(ex_gen)
Expand Down

0 comments on commit fc3fff4

Please sign in to comment.