arize-phoenix 0.0.18rc1__tar.gz → 0.0.19__tar.gz

{arize_phoenix-0.0.18rc1 → arize_phoenix-0.0.19}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: arize-phoenix
-Version: 0.0.18rc1
+Version: 0.0.19
 Summary: ML Observability in your notebook
 Project-URL: Documentation, https://docs.arize.com/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues

{arize_phoenix-0.0.18rc1 → arize_phoenix-0.0.19}/src/phoenix/__init__.py

@@ -7,7 +7,7 @@ from .datasets import (
 )
 from .session import Session, active_session, close_app, launch_app
 
-__version__ = "0.0.18rc1"
+__version__ = "0.0.19"
 
 # module level doc-string
 __doc__ = """

{arize_phoenix-0.0.18rc1 → arize_phoenix-0.0.19}/src/phoenix/metrics/timeseries.py

@@ -1,4 +1,4 @@
-from datetime import datetime, timedelta
+from datetime import datetime, timedelta, timezone
 from functools import partial
 from itertools import accumulate, repeat
 from typing import Any, Callable, Iterable, Iterator, Tuple, cast
@@ -195,9 +195,15 @@ def _results(
         # pandas row indexing is stop-exclusive
         row_slice = slice(row_start, row_stop)
         filtered = dataframe.iloc[row_slice, :]
-        yield filtered.groupby(
+        res = filtered.groupby(
             group,
             group_keys=True,
         ).apply(
             calculate_metrics,
-        ).loc[result_slice, :]
+        )
+
+        # NB: on ubuntu, we lose the timezone information when there is no data
+        if res.index.tzinfo is None:  # type: ignore
+            res = res.set_axis(res.index.tz_localize(timezone.utc), axis=0)  # type: ignore
+
+        yield res.loc[result_slice, :]

arize_phoenix-0.0.19/src/phoenix/server/api/input_types/TimeRange.py

@@ -0,0 +1,28 @@
+from datetime import datetime, timezone
+
+import strawberry
+
+
+@strawberry.input
+class TimeRange:
+    start: datetime = strawberry.field(
+        description="The start of the time range",
+    )
+    end: datetime = strawberry.field(
+        description="The end of the time range. Right exclusive.",
+    )
+
+    def __post_init__(self) -> None:
+        setattr(
+            self,
+            "start",
+            self.start.astimezone(timezone.utc),
+        )
+        setattr(
+            self,
+            "end",
+            self.end.astimezone(timezone.utc),
+        )
+
+    def is_valid(self) -> bool:
+        return self.start < self.end

arize_phoenix-0.0.18rc1/src/phoenix/server/api/input_types/TimeRange.py

@@ -1,12 +0,0 @@
-from datetime import datetime
-
-import strawberry
-
-
-@strawberry.input
-class TimeRange:
-    start: datetime = strawberry.field(description="The start of the time range")
-    end: datetime = strawberry.field(description="The end of the time range. Right exclusive.")
-
-    def is_valid(self) -> bool:
-        return self.start < self.end