Skip to content

Commit

Permalink
[Tests] Fix frames client getter [1.5.x]
Browse files Browse the repository at this point in the history
Fix the backport introduced in mlrun#4564.
This method was changed in 1.6.x and does not work in 1.5.x the same.
I tested it locally with 1.5.x, and it works now.
  • Loading branch information
jond01 committed Nov 13, 2023
1 parent c59725f commit 56067e7
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions tests/system/model_monitoring/test_model_monitoring.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@
import mlrun.serving.routers
from mlrun.errors import MLRunNotFoundError
from mlrun.model import BaseMetadata
from mlrun.model_monitoring.writer import _TSDB_BE, ModelMonitoringWriter
from mlrun.model_monitoring.writer import _TSDB_BE
from mlrun.runtimes import BaseRuntime
from mlrun.utils.v3io_clients import get_frames_client
from tests.system.base import TestMLRunSystem
Expand Down Expand Up @@ -1004,8 +1004,9 @@ def _log_model(self) -> str:

@classmethod
def _test_v3io_tsdb_record(cls) -> None:
frames = ModelMonitoringWriter._get_v3io_frames_client(
v3io_container="users",
frames = mlrun.utils.v3io_clients.get_frames_client(
address=mlrun.mlconf.v3io_framesd,
container="users",
)
df: pd.DataFrame = frames.read(
backend=_TSDB_BE,
Expand Down

0 comments on commit 56067e7

Please sign in to comment.