Skip to content

Commit

Permalink
make message handler logging levels consistent
Browse files Browse the repository at this point in the history
Summary: Messages are also now info so that runner can tell things are happening. Debug is saved for very verbose errors.

Test Plan:
  • Loading branch information
JasonKChow committed Feb 7, 2025
1 parent cbdb46f commit 5bd89c7
Show file tree
Hide file tree
Showing 9 changed files with 9 additions and 9 deletions.
2 changes: 1 addition & 1 deletion aepsych/database/db.py
Original file line number Diff line number Diff line change
Expand Up @@ -321,7 +321,7 @@ def record_setup(

record.timestamp = datetime.datetime.now()
record.parent = master_table
logger.debug(f"record_setup = [{record}]")
logger.debug(f"replay_record_setup = [{record}]")

self._add_commit(session, record)

Expand Down
2 changes: 1 addition & 1 deletion aepsych/server/message_handlers/handle_ask.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ def handle_ask(server, request):
"config" -- dictionary with config (keys are strings, values are floats)
"is_finished" -- bool, true if the strat is finished
"""
logger.debug("got ask message!")
logger.info("got ask message!")
if server._pregen_asks:
params = server._pregen_asks.pop()
else:
Expand Down
2 changes: 1 addition & 1 deletion aepsych/server/message_handlers/handle_can_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
def handle_can_model(server, request):
# Check if the strategy has finished initialization; i.e.,
# if it has a model and data to fit (strat.can_fit)
logger.debug("got can_model message!")
logger.info("got can_model message!")
if not server.is_performing_replay:
server.db.record_message(
master_table=server._db_master_record, type="can_model", request=request
Expand Down
2 changes: 1 addition & 1 deletion aepsych/server/message_handlers/handle_info.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ def handle_info(server, request: Dict[str, Any]) -> Dict[str, Any]:
Returns:
Dict[str, Any]: Returns dictionary containing the current state of the experiment
"""
logger.debug("got info message!")
logger.info("got info message!")

ret_val = info(server)

Expand Down
2 changes: 1 addition & 1 deletion aepsych/server/message_handlers/handle_params.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@


def handle_params(server, request):
logger.debug("got parameters message!")
logger.info("got parameters message!")
if not server.is_performing_replay:
server.db.record_message(
master_table=server._db_master_record, type="parameters", request=request
Expand Down
2 changes: 1 addition & 1 deletion aepsych/server/message_handlers/handle_query.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@


def handle_query(server, request):
logger.debug("got query message!")
logger.info("got query message!")
if not server.is_performing_replay:
server.db.record_message(
master_table=server._db_master_record, type="query", request=request
Expand Down
2 changes: 1 addition & 1 deletion aepsych/server/message_handlers/handle_resume.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@


def handle_resume(server, request):
logger.debug("got resume message!")
logger.info("got resume message!")
strat_id = int(request["message"]["strat_id"])
server.strat_id = strat_id
if not server.is_performing_replay:
Expand Down
2 changes: 1 addition & 1 deletion aepsych/server/message_handlers/handle_setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,7 @@ def configure(server, config=None, **config_args):


def handle_setup(server, request):
logger.debug("got setup message!")
logger.info("got setup message!")
### make a temporary config object to derive parameters because server handles config after table
if (
"config_str" in request["message"].keys()
Expand Down
2 changes: 1 addition & 1 deletion aepsych/server/message_handlers/handle_tell.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@


def handle_tell(server, request):
logger.debug("got tell message!")
logger.info("got tell message!")

if not server.is_performing_replay:
server.db.record_message(
Expand Down

0 comments on commit 5bd89c7

Please sign in to comment.