From 2382067d91b5cba67ce90817925e6e70bc84f25a Mon Sep 17 00:00:00 2001
From: naweedkhan <naweed.khan@ibm.com>
Date: Wed, 1 Mar 2023 16:04:36 +0200
Subject: [PATCH] ADD: train with a single loss

- a list of losses no longer needs to be specified

Signed-off-by: naweedkhan <naweed.khan@ibm.com>
---
 lnn/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lnn/model.py b/lnn/model.py
index 1f0d64f..5816c27 100644
--- a/lnn/model.py
+++ b/lnn/model.py
@@ -670,7 +670,7 @@ def loss_fn(self, losses):
                 f"expected losses from the following {[l.name for l in Loss]}"
             )
         elif isinstance(losses, Loss):
-            losses = [losses]
+            losses = {losses: None}
         elif isinstance(losses, list):
             losses = {c: None for c in losses}
         result = list()