update test, fix where clause

Arize-ai · Oct 30, 2024 · cc59302 · cc59302
1 parent b4a2089
commit cc59302
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -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)