From 3a2353893d44890ccf91c80236776babe25e8b0a Mon Sep 17 00:00:00 2001 From: polyaxon-ci Date: Sat, 21 Oct 2023 18:01:15 +0200 Subject: [PATCH] Update imports --- traceml/traceml/tracking/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/traceml/traceml/tracking/__init__.py b/traceml/traceml/tracking/__init__.py index a66cdc434..25b8aa08e 100644 --- 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