clean up

Arize-ai · Oct 11, 2024 · 30bc2bd · 30bc2bd
1 parent fb9817e
commit 30bc2bd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/phoenix/db/models.py b/src/phoenix/db/models.py
@@ -491,9 +491,9 @@ def _example_count(cls) -> ColumnElement[int]:
             .label("example_count")
         )
 
-    async def load_example_count(self, conversation: AsyncSession) -> None:
+    async def load_example_count(self, session: AsyncSession) -> None:
         if not hasattr(self, "_example_count_value"):
-            self._example_count_value = await conversation.scalar(
+            self._example_count_value = await session.scalar(
                 select(
                     func.sum(
                         case(