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)