From c7b13ed08c4d13910fc6b05b3d79d5e84319b864 Mon Sep 17 00:00:00 2001 From: mayankbajoria Date: Fri, 17 Jul 2026 11:01:11 +0530 Subject: [PATCH] Fix logger config for datastores without _config --- neural_lam/utils/logging.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/neural_lam/utils/logging.py b/neural_lam/utils/logging.py index 706629e0..379e3da6 100644 --- a/neural_lam/utils/logging.py +++ b/neural_lam/utils/logging.py @@ -117,7 +117,7 @@ def setup_training_logger( return pl.loggers.WandbLogger( project=args.logger_project, name=None if args.wandb_id else run_name, - config=dict(training=vars(args), datastore=datastore._config), + config=dict(training=vars(args), datastore=getattr(datastore, "_config", None)), resume=wandb_resume, id=args.wandb_id, save_dir=run_dir, @@ -140,7 +140,7 @@ def setup_training_logger( save_dir=run_dir, ) training_logger.log_hyperparams( - dict(training=vars(args), datastore=datastore._config) + dict(training=vars(args), datastore=getattr(datastore, "_config", None)) ) return training_logger else: