diff --git a/src/otg/method/l2g/model.py b/src/otg/method/l2g/model.py index 7429c6770..23405e6b8 100644 --- a/src/otg/method/l2g/model.py +++ b/src/otg/method/l2g/model.py @@ -136,7 +136,7 @@ def log_to_wandb( wandb_run.log(gs_counts_dict) # Missingness rates wandb_run.log( - "missingnessRates", training_data.calculate_feature_missingness_rate() + {"missingnessRates": training_data.calculate_feature_missingness_rate()} ) @classmethod