Fix ignite init process

polyaxon · Nov 29, 2023 · 05ef428 · 05ef428
1 parent 1d1419b
commit 05ef428
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/traceml/traceml/integrations/ignite.py b/traceml/traceml/integrations/ignite.py
@@ -9,4 +9,6 @@
 
 class Logger(PolyaxonLogger):
     def __init__(self, *args, **kwargs):
-        self.experiment = kwargs.get("run", Run(*args, **kwargs))
+        self.experiment = kwargs.get("run")
+        if not self.experiment:
+            self.experiment = Run(*args, **kwargs)