Update imports

polyaxon · Oct 21, 2023 · 3a23538 · 3a23538
1 parent f393cb1
commit 3a23538
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/traceml/traceml/tracking/__init__.py b/traceml/traceml/tracking/__init__.py
@@ -1,7 +1,8 @@
 from datetime import datetime
 from typing import Any, Dict, List, Optional, Sequence, Union
 
-from polyaxon.client import *
+from polyaxon._client.run import RunClient
+from polyaxon._sdk.schemas import V1Run
 from traceml.artifacts import V1ArtifactKind, V1RunArtifact
 from traceml.tracking.run import Run