arize-phoenix 0.0.18rc1__py3-none-any.whl → 0.0.19__py3-none-any.whl

{arize_phoenix-0.0.18rc1.dist-info → arize_phoenix-0.0.19.dist-info}/METADATA

@@ -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.dist-info → arize_phoenix-0.0.19.dist-info}/RECORD

@@ -1,4 +1,4 @@
-phoenix/__init__.py,sha256=lq7jUqia25zHlxg-WkgoniqAFp9CHlqTrlkIw2CrHSg,1000
+phoenix/__init__.py,sha256=MS-mFdJlXRrlUlXo7KWlnmNRpais546_SbhjGXBcCDU,997
 phoenix/config.py,sha256=tjNn9oqDxQmeO85sCchLlTsDiRJ6AoK0CTt_Uc_hrKM,1442
 phoenix/py.typed,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 phoenix/services.py,sha256=pamdcZ9hp9BqgRV0ZIcMsjBYp8HCpIQPdxaXQfQLxYU,3711
@@ -26,7 +26,7 @@ phoenix/metrics/median.py,sha256=M-d00yh-JVodi7QC-smA6jbAgjbWajYHlOEBKAWk0IY,119
 phoenix/metrics/metrics.py,sha256=3ggWwiVR5dmNqsEDuJvJiLgttkHlnHK3uB7IBqjLwwk,7292
 phoenix/metrics/mixins.py,sha256=zk-6Il1asTbMP_hwvfXABPu_8jYynQyEudDIcWRhU0c,5727
 phoenix/metrics/percent_empty.py,sha256=0pRA-_nFqGgYfTnxe_uIZX9RQV-O1ADzh6KQZIbsXnk,465
-phoenix/metrics/timeseries.py,sha256=bMshh7OJCD95VGelPz1MMj6KMePjKYkAuRsk4xtMu_A,6609
+phoenix/metrics/timeseries.py,sha256=20WA1DN79NDc2iB7xm71XAK4Sn4pwXANEo1skr9pz_I,6864
 phoenix/pointcloud/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 phoenix/pointcloud/clustering.py,sha256=IzcG67kJ2hPP7pcqVmKPSL_6gKRonKdOT3bCtbTOqnk,820
 phoenix/pointcloud/pointcloud.py,sha256=als0aitTA__4PrSqBk6qPPKe8fIG-ZSnlVRVkfMorBU,2290
@@ -41,7 +41,7 @@ phoenix/server/api/interceptor.py,sha256=pivESQdLQAWrMjToK7b8rl52NobiatqrWnK8Qe0
 phoenix/server/api/schema.py,sha256=QYONWcS3xNvqD6t2hp63WkoZIQVrW9kH59MkVFXf3_I,1143
 phoenix/server/api/input_types/DimensionInput.py,sha256=Vfx5FmiMKey4-EHDQsQRPzSAMRJMN5oVMLDUl4NKAa8,164
 phoenix/server/api/input_types/Granularity.py,sha256=6SVfZ5yTZYq1PI6vdpjfkBUc4YilLSkF-k6okuSNbbQ,2301
-phoenix/server/api/input_types/TimeRange.py,sha256=8GhSVyFC3byuvpcOG2lhC5ZKXgXW0g_UtaVdDPAfxwk,334
+phoenix/server/api/input_types/TimeRange.py,sha256=cKXGQCuKtc-MwnaC_HBhfVnrM-f6i98JIiJPdQ2Gb64,638
 phoenix/server/api/input_types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/server/api/types/DataQualityMetric.py,sha256=eq2W8TutOdoeWcRWRUSTNBKR9eP0lfZUb1tlqYJ6flw,243
 phoenix/server/api/types/Dataset.py,sha256=YIiy6sU6CnH8iiBZXJmMDjKVlSlupOysmFoLOnvzGYE,2928
@@ -85,8 +85,8 @@ phoenix/session/__init__.py,sha256=qFZaChQ3SwNAOR-LD4hzQQthe32haw1vU1JeBvgMnRQ,1
 phoenix/session/session.py,sha256=qVg8q6DG-YotNVWXIywdR2q0VgRYsV7GiDMyvE8rje4,8650
 phoenix/utils/__init__.py,sha256=alIDGBnxWH4JvP-UW-7N99seBBi0r1GV1h8f1ERFBec,21
 phoenix/utils/utils.py,sha256=hZK3a_nLFYiQb1O6EcMF3mVmhTjnfuJ5WMcjhvOu7zk,427
-arize_phoenix-0.0.18rc1.dist-info/METADATA,sha256=8HODB7f8GLiovOxKdix7h2lYzf4dY-sA0Lv-fJfIUqg,10762
-arize_phoenix-0.0.18rc1.dist-info/WHEEL,sha256=Fd6mP6ydyRguakwUJ05oBE7fh2IPxgtDN9IwHJ9OqJQ,87
-arize_phoenix-0.0.18rc1.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
-arize_phoenix-0.0.18rc1.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
-arize_phoenix-0.0.18rc1.dist-info/RECORD,,
+arize_phoenix-0.0.19.dist-info/METADATA,sha256=cj9jXkDT7cR0BX4SehkMuhTbWgYNDeJJaeVuXQpxyng,10759
+arize_phoenix-0.0.19.dist-info/WHEEL,sha256=Fd6mP6ydyRguakwUJ05oBE7fh2IPxgtDN9IwHJ9OqJQ,87
+arize_phoenix-0.0.19.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
+arize_phoenix-0.0.19.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
+arize_phoenix-0.0.19.dist-info/RECORD,,

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__ = """

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, :]

phoenix/server/api/input_types/TimeRange.py

@@ -1,12 +1,28 @@
-from datetime import datetime
+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.")
+    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