Update traceml exception

polyaxon · Oct 5, 2023 · 33b2792 · 33b2792
1 parent 0cab3dd
commit 33b2792
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 7 deletions.
diff --git a/traceml/traceml/exceptions.py b/traceml/traceml/exceptions.py
@@ -1,5 +1,3 @@
 from polyaxon.exceptions import PolyaxonClientException
 
-
-class TracemlException(PolyaxonClientException):
-    pass
+TracemlException = PolyaxonClientException
diff --git a/traceml/traceml/integrations/tensorboard.py b/traceml/traceml/integrations/tensorboard.py
@@ -2,7 +2,6 @@
 
 from typing import TYPE_CHECKING, Any, Optional
 
-from polyaxon.exceptions import PolyaxonClientException
 from traceml import tracking
 from traceml.exceptions import TracemlException
 from traceml.logger import logger
@@ -64,9 +63,7 @@ def process_summary(
                     log_histo=log_histo,
                     log_tensor=log_tensor,
                 )
-            except PolyaxonClientException(
-                "Polyaxon failed processing tensorboard summary."
-            ):
+            except TracemlException("Polyaxon failed processing tensorboard summary."):
                 pass
 
     @classmethod