From cc59302f6047486b46fa7aa266626c182421555f Mon Sep 17 00:00:00 2001 From: Parker Stafford Date: Wed, 30 Oct 2024 14:07:16 -0700 Subject: [PATCH] update test, fix where clause --- src/phoenix/server/api/types/ProjectSession.py | 2 +- tests/unit/server/api/dataloaders/test_latency_ms_quantiles.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/src/phoenix/server/api/types/ProjectSession.py b/src/phoenix/server/api/types/ProjectSession.py index 2fbdd6dfdc..14b8ac49cc 100644 --- a/src/phoenix/server/api/types/ProjectSession.py +++ b/src/phoenix/server/api/types/ProjectSession.py @@ -123,7 +123,7 @@ async def trace_latency_ms_quantile( None, None, probability, - models.Trace.project_session_rowid == self.project_rowid, + models.Trace.project_session_rowid == self.id_attr, ) ) diff --git a/tests/unit/server/api/dataloaders/test_latency_ms_quantiles.py b/tests/unit/server/api/dataloaders/test_latency_ms_quantiles.py index 65ce6be50a..4481aaf33c 100644 --- a/tests/unit/server/api/dataloaders/test_latency_ms_quantiles.py +++ b/tests/unit/server/api/dataloaders/test_latency_ms_quantiles.py @@ -57,6 +57,7 @@ async def test_latency_ms_quantiles_p25_p50_p75( TimeRange(start=start_time, end=end_time), "'_5_' in name" if kind == "span" else None, probability, + None, ) for kind in kinds for id_ in range(10)