From 301b8164635bb9b9a32420481cd4ba67a1d0fc8f Mon Sep 17 00:00:00 2001
From: Nikhil Singh <124866156+Ritinikhil@users.noreply.github.com>
Date: Fri, 20 Mar 2026 15:39:15 +0530
Subject: [PATCH] Handle unsupported logger types in training logger

Raise ValueError for unsupported logger types.
---
 neural_lam/utils.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/neural_lam/utils.py b/neural_lam/utils.py
index 742ef9823..f386ff286 100644
--- a/neural_lam/utils.py
+++ b/neural_lam/utils.py
@@ -540,8 +540,12 @@ def setup_training_logger(datastore, args, run_name):
         training_logger.log_hyperparams(
             dict(training=vars(args), datastore=datastore._config)
         )
-
-    return training_logger
+        return training_logger
+    else:
+        raise ValueError(
+            f"Unsupported logger type: {args.logger!r}. "
+            "Supported loggers are: 'wandb', 'mlflow'."
+        )
 
 
 def inverse_softplus(x, beta=1, threshold=20):