From 05ef428c2c03efe4653d098eecf2ddef17306649 Mon Sep 17 00:00:00 2001 From: polyaxon-ci Date: Wed, 29 Nov 2023 01:31:49 +0100 Subject: [PATCH] Fix ignite init process --- traceml/traceml/integrations/ignite.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/traceml/traceml/integrations/ignite.py b/traceml/traceml/integrations/ignite.py index 25fae8e06..e9f4e4b95 100644 --- 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)