arize-phoenix 0.0.24rc0__py3-none-any.whl → 0.0.24rc2__py3-none-any.whl

{arize_phoenix-0.0.24rc0.dist-info → arize_phoenix-0.0.24rc2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: arize-phoenix
-Version: 0.0.24rc0
+Version: 0.0.24rc2
 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.24rc0.dist-info → arize_phoenix-0.0.24rc2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-phoenix/__init__.py,sha256=PIS-tjbGiHDur05VJ2Qbnt1fCvP1QjJj1HpY1XYvp1k,1049
+phoenix/__init__.py,sha256=MDPqkLU2t2D4gPOm1SNfCt1YaQnGF57LQcyGPQRokRk,1049
 phoenix/config.py,sha256=tjNn9oqDxQmeO85sCchLlTsDiRJ6AoK0CTt_Uc_hrKM,1442
 phoenix/py.typed,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 phoenix/services.py,sha256=pamdcZ9hp9BqgRV0ZIcMsjBYp8HCpIQPdxaXQfQLxYU,3711
@@ -8,10 +8,10 @@ phoenix/core/dimension_data_type.py,sha256=FEBp4p06LlFpAXq-ftxDAFymBtU_pYTFmJjFc
 phoenix/core/dimension_type.py,sha256=EKFmPzuGr1cn7t4vD-XDk9s836j6U3iRbSu3Z2sO8sM,171
 phoenix/core/embedding_dimension.py,sha256=zKGbcvwOXgLf-yrJBpQyKtd-LEOPRKHnUToyAU8Owis,87
 phoenix/core/model.py,sha256=JLAw1kc8Ctrm99E7Nb7eOWghjC9O_V1AFu-e5byD5UY,4638
-phoenix/core/model_schema.py,sha256=MAh_OxZDp8XUWMjuXsfwRmPxPZF7JNKlbP1ZYohDMzU,47447
-phoenix/core/model_schema_adapter.py,sha256=Tf49ZKcX33vcsatQ7CW4s31cejJd6xAZMfCq9vv1Chs,5522
+phoenix/core/model_schema.py,sha256=nHmNbmfFnJL4a2o40YxE7BL-S5FLGeS4waO_8Jn2W1w,47340
+phoenix/core/model_schema_adapter.py,sha256=Kqr1doD12xtujTtWX-zgPu1HfRcdM1756ZCa3uW0f_Y,7165
 phoenix/datasets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-phoenix/datasets/dataset.py,sha256=zKJV_YHBgcObioatpBsY8SXhEjBLXBiW27eObVoTps4,21682
+phoenix/datasets/dataset.py,sha256=J92RSwlbrh-Nj6BwQkO0GsIdSe81MuHr1xz8hwf69GY,21824
 phoenix/datasets/errors.py,sha256=-Iyk8rsvP_KX-P4gOqjm26slkDq1-9CohK07_LkrYCI,8117
 phoenix/datasets/fixtures.py,sha256=9rmJgsF50vmuwR7PavNf7kdFsLQiasta6iJEzh7GvkM,15954
 phoenix/datasets/schema.py,sha256=1sSdtKkN7L61rsb8FNeTxJZG-ysGR7JGgT08WKGuDFs,3658
@@ -26,7 +26,7 @@ phoenix/metrics/wrappers.py,sha256=fCWIcIp30-YZmlzL4hpv2u07QRfCNn4WtI0mcQN0db8,8
 phoenix/pointcloud/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 phoenix/pointcloud/clustering.py,sha256=IzcG67kJ2hPP7pcqVmKPSL_6gKRonKdOT3bCtbTOqnk,820
 phoenix/pointcloud/pointcloud.py,sha256=Ntj5juwMfwjPqyacPNGyH0lsItqpMTqtKiDdIEufnTY,2196
-phoenix/pointcloud/projectors.py,sha256=-FRYi420TrJkibFcMwAsXa8HK4u6EQRxHKslZpeldkw,716
+phoenix/pointcloud/projectors.py,sha256=zO_RrtDYSv2rqVOfIP2_9Cv11Dc8EmcZR94xhFcBYPU,1057
 phoenix/server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/server/app.py,sha256=kWG5YWSnzZ668aA9TNoIeoc6Gxf6fYCgab6ZDAdDsig,4013
 phoenix/server/main.py,sha256=PUK042dTYCgh18eELtPhldBujUxQLAmrMDJD2Oz8Exk,3035
@@ -63,10 +63,10 @@ phoenix/server/api/types/ExportEventsMutation.py,sha256=NctthWhHFR4wKva1jfyWX4-5
 phoenix/server/api/types/ExportedFile.py,sha256=e3GTn7B5LgsTbqiwjhMCQH7VsiqXitrBO4aCMS1lHsg,163
 phoenix/server/api/types/Model.py,sha256=I7liNDDsl1cykJjMQutUI18ZiQ5EXBar3I0a_-C66kw,7228
 phoenix/server/api/types/NumericRange.py,sha256=afEjgF97Go_OvmjMggbPBt-zGM8IONewAyEiKEHRds0,192
-phoenix/server/api/types/PerformanceMetric.py,sha256=_jUkpLCUHP3bl6224Yt_7FOuHG2CS0GET1qajzayjC0,351
+phoenix/server/api/types/PerformanceMetric.py,sha256=W92B7OghEOgzFvmY0LCqpgavHaQggTGshdgfD0yqHX4,350
 phoenix/server/api/types/PromptResponse.py,sha256=Q8HKtpp8GpUOcxPCzZpkkokidDd6u0aZOv_SuPZZd5Q,630
 phoenix/server/api/types/ScalarDriftMetricEnum.py,sha256=IUAcRPpgL41WdoIgK6cNk2Te38SspXGyEs-S1fY23_A,232
-phoenix/server/api/types/Segments.py,sha256=J5g5sOP74-aW7F9sR3H1EMpRyY4L4QoG8KRT9bUFb_Y,2174
+phoenix/server/api/types/Segments.py,sha256=eUoisp10Y1dCFCSphOjNA52g2Ja2chrvc_q1yMm-b2k,2903
 phoenix/server/api/types/TimeSeries.py,sha256=pcSjoPLNq3fNiCt6amiZEGtLDF14HPP7HVXA8znvFmQ,5190
 phoenix/server/api/types/UMAPPoints.py,sha256=GS37oRtFfEqucgMxXp0RtoZaz4D5Ye_lSYQYR3JlMe4,1451
 phoenix/server/api/types/VectorDriftMetricEnum.py,sha256=etiJM5ZjQuD-oE7sY-FbdIKY050jk3IU49IMwmfJbEc,188
@@ -84,12 +84,12 @@ phoenix/server/static/apple-touch-icon.png,sha256=fOfpjqGpWYbJ0eAurKsyoZP1EAs6ZV
 phoenix/server/static/favicon.ico,sha256=bY0vvCKRftemZfPShwZtE93DiiQdaYaozkPGwNFr6H8,34494
 phoenix/server/static/index.css,sha256=jeV8eWPiHUcUmb_0yp_rI1odc-RnxlXVgMT-x9HGWbo,1817
 phoenix/server/static/index.html,sha256=xPZZH-y4dWlbDutPEV1k0rhmWJtIV-Db9aYP-dEc7wM,703
-phoenix/server/static/index.js,sha256=7qQbwDFSRuf1UGGb7Mir60p9E6cF3prq3bGwKhErx7s,2338799
+phoenix/server/static/index.js,sha256=YtwDNwj3GObOQYv8GllsZZ3Q_0Is3aZRQKRp8SWFtJU,2358860
 phoenix/server/static/modernizr.js,sha256=mvK-XtkNqjOral-QvzoqsyOMECXIMu5BQwSVN_wcU9c,2564
 phoenix/session/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/session/session.py,sha256=sHm0qoPexNBF49a6VlGw0Oi81Zkj_EjrJCY_b3VK-Yo,8658
-arize_phoenix-0.0.24rc0.dist-info/METADATA,sha256=SuZHDTY1IiqWo8d1Vey5DIfrGN60AQNj6hpCIpRz0wk,10813
-arize_phoenix-0.0.24rc0.dist-info/WHEEL,sha256=Fd6mP6ydyRguakwUJ05oBE7fh2IPxgtDN9IwHJ9OqJQ,87
-arize_phoenix-0.0.24rc0.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
-arize_phoenix-0.0.24rc0.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
-arize_phoenix-0.0.24rc0.dist-info/RECORD,,
+arize_phoenix-0.0.24rc2.dist-info/METADATA,sha256=aNkZu5EL2zd-5k11SJnUF1JnoMTYdn9T4V_A0oxa6SM,10813
+arize_phoenix-0.0.24rc2.dist-info/WHEEL,sha256=9QBuHhg6FNW7lppboF2vKVbCGTVzsFykgRQjjlajrhA,87
+arize_phoenix-0.0.24rc2.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
+arize_phoenix-0.0.24rc2.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
+arize_phoenix-0.0.24rc2.dist-info/RECORD,,

{arize_phoenix-0.0.24rc0.dist-info → arize_phoenix-0.0.24rc2.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.13.0
+Generator: hatchling 1.18.0
 Root-Is-Purelib: true
 Tag: py3-none-any

phoenix/__init__.py

@@ -3,7 +3,7 @@ from .datasets.fixtures import ExampleDatasets, load_example
 from .datasets.schema import EmbeddingColumnNames, Schema
 from .session.session import Session, active_session, close_app, launch_app
 
-__version__ = "0.0.24rc0"
+__version__ = "0.0.24rc2"
 
 # module level doc-string
 __doc__ = """

phoenix/core/model_schema.py

@@ -599,10 +599,8 @@ class Events(ModelData):
     @cached_property
     def time_range(self) -> TimeRange:
         if self._self_model is None or self.empty:
-            # NOTE: as of Python 3.8.16, pandas 1.5.3:
-            # >>> isinstance(pd.NaT, datetime.datetime)
-            # True
-            return TimeRange(pd.NaT, pd.NaT)  # type: ignore
+            now = datetime.now(timezone.utc)
+            return TimeRange(now, now)
         model = cast(Model, self._self_model)
         min_max = _agg_min_max(model[TIMESTAMP](self))
         start_time = cast(datetime, min_max.min())

phoenix/core/model_schema_adapter.py

@@ -1,7 +1,9 @@
 from itertools import chain
-from typing import Dict, Iterable, List, Optional, Tuple
+from operator import itemgetter
+from typing import Dict, Iterable, List, Optional, Sized, Tuple
 
 import pandas as pd
+from pandas.api.types import is_object_dtype
 from typing_extensions import TypeAlias, TypeGuard
 
 from phoenix import Dataset, EmbeddingColumnNames
@@ -107,8 +109,19 @@ def create_model_from_datasets(*datasets: Optional[Dataset]) -> Model:
         prediction_score=_take_first_str(prediction_scores),
         actual_label=_take_first_str(actual_labels),
         actual_score=_take_first_str(actual_scores),
-        features=chain(features, translated_embeddings),
-        tags=tags,
+        features=chain(
+            *_split_vectors_vs_scalars(
+                features,
+                *map(itemgetter(1), named_dataframes),
+            ),
+            translated_embeddings,
+        ),
+        tags=chain(
+            *_split_vectors_vs_scalars(
+                tags,
+                *map(itemgetter(1), named_dataframes),
+            )
+        ),
         prompt=next(map(_translate_embedding, prompts), None),
         response=next(map(_translate_embedding, responses), None),
     )(
@@ -137,3 +150,37 @@ def _translate_embedding(
         link_to_data=embedding.link_to_data_column_name,
         display_name=display_name,
     )
+
+
+def _split_vectors_vs_scalars(
+    names: Iterable[str],
+    *dataframes: pd.DataFrame,
+) -> Tuple[List[str], List[Embedding]]:
+    """A best-effort attempt at separating vector columns from scalar columns
+    by examining the first non-null item of the column from each dataframe. If
+    any item is `Iterable` and `Sized`, but not `str`, then the column is
+    returned as `Embedding`, else it's returned as scalar.
+    """
+    scalars: List[str] = []
+    vectors: List[Embedding] = []
+    # convert to sets for a speedier lookup
+    column_names = [set(df.columns) for df in dataframes]
+    for name in names:
+        for i, df in enumerate(dataframes):
+            if df.empty or name not in column_names[i]:
+                continue
+            series = df.loc[:, name]
+            if not is_object_dtype(series):
+                continue
+            item = series.iat[series.isna().argmin()]
+            if (
+                isinstance(item, str)  # str is scalar, but Iterable
+                or not isinstance(item, Iterable)
+                or not isinstance(item, Sized)
+            ):
+                continue
+            vectors.append(Embedding(vector=name))
+            break
+        else:
+            scalars.append(name)
+    return scalars, vectors

phoenix/datasets/dataset.py

@@ -443,7 +443,11 @@ def _normalize_timestamps(
     if (timestamp_column_name := schema.timestamp_column_name) is None:
         timestamp_column_name = "timestamp"
         schema = replace(schema, timestamp_column_name=timestamp_column_name)
-        timestamp_column = Series([default_timestamp] * len(dataframe), index=dataframe.index)
+        timestamp_column = (
+            Series([default_timestamp] * len(dataframe), index=dataframe.index)
+            if len(dataframe)
+            else Series([default_timestamp]).iloc[:0].set_axis(dataframe.index, axis=0)
+        )
     elif is_numeric_dtype(timestamp_column_dtype := dataframe[timestamp_column_name].dtype):
         timestamp_column = to_datetime(dataframe[timestamp_column_name], unit="s", utc=True)
     elif is_datetime64tz_dtype(timestamp_column_dtype):

phoenix/pointcloud/projectors.py

@@ -11,6 +11,7 @@ with warnings.catch_warnings():
 
     warnings.simplefilter("ignore", category=NumbaWarning)
     from umap import UMAP
+
 Matrix: TypeAlias = npt.NDArray[np.float64]
 
 
@@ -24,4 +25,11 @@ class Umap:
     min_dist: float = 0.1
 
     def project(self, mat: Matrix, n_components: int) -> Matrix:
-        return _center(UMAP(**asdict(self), n_components=n_components).fit_transform(mat))
+        config = asdict(self)
+        config["n_components"] = n_components
+        if len(mat) <= n_components:
+            # init='spectral', the default, cannot be used when n_components
+            # is greater or equal to the number of samples.
+            # see https://github.com/lmcinnes/umap/issues/201#issuecomment-462097103
+            config["init"] = "random"
+        return _center(UMAP(**config).fit_transform(mat))

phoenix/server/api/types/PerformanceMetric.py

@@ -10,4 +10,4 @@ from phoenix.metrics.wrappers import SkEval
 class PerformanceMetric(Enum):
     # To become enum values, functions need to be wrapped in partial.
     # See https://stackoverflow.com/a/40339397
-    accuracy_score = partial(SkEval.accuracy_score)  # type: ignore
+    accuracyScore = partial(SkEval.accuracy_score)  # type: ignore

phoenix/server/api/types/Segments.py

@@ -84,5 +84,19 @@ class Segments:
     )
 
     def append(self, other: Segment) -> None:
+        if (
+            isinstance(other.bin, IntervalBin)
+            and not math.isfinite(other.bin.range.start)
+            and not math.isfinite(other.bin.range.end)
+            and other.counts.primary_value == 0
+            and other.counts.reference_value == 0
+        ):
+            # Skip the interval bin with zero counts if it has -inf and +inf
+            # as the endpoints, as it could occur due to all values being
+            # missing. Because such bin can exist a priori, the caller may
+            # still try to append it, but the bin has no real value at this
+            # point (and can cause problems for graphql because of the end
+            # points are not serializable to JSON).
+            return
         self.segments.append(other)
         self.total_counts += other.counts