Ensure additional metadata to trackers don't throw error in happy cas…

…e. (#290)

Signed-off-by: Dushyant Behl <[email protected]>

tuning/sft_trainer.py

@@ -337,6 +337,7 @@ def train(
             try:
                 for k, v in additional_metrics.items():
                     tracker.track(metric=v, name=k, stage="additional_metrics")
+                if exp_metadata:
                     tracker.set_params(params=exp_metadata, name="experiment_metadata")
             except ValueError as e:
                 logger.error(

tuning/trackers/aimstack_tracker.py

@@ -172,7 +172,9 @@ def set_params(self, params, name="extra_params"):
         Raises:
             ValueError: the params passed is None or not of type dict
         """
-        if params is None or (not isinstance(params, dict)):
+        if params is None:
+            return
+        if not isinstance(params, dict):
             raise ValueError(
                 "set_params passed to aimstack should be called with a dict of params"
             )