arize-phoenix 0.0.2rc3__tar.gz → 0.0.2rc5__tar.gz

{arize_phoenix-0.0.2rc3 → arize_phoenix-0.0.2rc5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: arize-phoenix
-Version: 0.0.2rc3
+Version: 0.0.2rc5
 Summary: ML Observability in your notebook
 Project-URL: Documentation, https://github.com/Arize-ai/phoenix#readme
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues
@@ -65,12 +65,12 @@ Phoenix provides MLOps insights at lightning speed with zero-config observabilit
 
 **_Phoenix is under active development. APIs may change at any time._**
 
-- [Installation](#installation)
-- [Getting Started](#getting-started)
-- [Documentation](#documentation)
-- [Community](#community)
-- [Contributing](#contributing)
-- [License](#license)
+-   [Installation](#installation)
+-   [Getting Started](#getting-started)
+-   [Documentation](#documentation)
+-   [Community](#community)
+-   [Contributing](#contributing)
+-   [License](#license)
 
 ## Installation
 
@@ -87,8 +87,9 @@ After installing `arize-phoenix` in your Jupyter or Colab environment, open your
 ```python
 import phoenix as px
 
-train_ds, prod_ds = px.load_dataset("sentiment_classification_language_drift")
-px.launch_app(train_ds, prod_ds)
+datasets = px.load_datasets("sentiment_classification_language_drift")
+session = px.launch_app(datasets.primary, datasets.reference)
+session.view()
 ```
 
 Next, visualize your embeddings and inspect problematic clusters of your production data.
@@ -96,6 +97,7 @@ Next, visualize your embeddings and inspect problematic clusters of your product
 TODO(#297): Include GIF where we navigate to embeddings, zoom in and rotate, and select a cluster.
 
 Don't forget to close the app when you're done.
+
 ```
 px.close_app()
 ```
@@ -109,21 +111,23 @@ For in-depth examples and explanations, read the [docs](https://docs.arize.com/p
 ## Community
 
 Join our community to connect with thousands of machine learning practitioners and ML observability enthusiasts.
-- 🌍 Join our [Slack community](https://join.slack.com/t/arize-ai/shared_invite/zt-1px8dcmlf-fmThhDFD_V_48oU7ALan4Q).
-- 💡 Ask questions and provide feedback in the *#phoenix-support* channel.
-- 🌟 Leave a star on our [GitHub](https://github.com/Arize-ai/phoenix).
-- 🐞 Report bugs with [GitHub Issues](https://github.com/Arize-ai/phoenix/issues).
-- 🗺️ Check out our [roadmap](https://github.com/orgs/Arize-ai/projects/45) to see where we're heading next.
-- 🎓 Learn the fundamentals of ML observability with our [introductory](https://arize.com/ml-observability-fundamentals/) and [advanced](https://arize.com/blog-course/) courses.
-- ✏️ Check out our [blog](https://arize.com/blog/). TODO(#291): Add blog filter for Phoenix
-- ✉️ Subscribe to our mailing list. TODO(#294): Add link
-- 🐦 Follow us on [Twitter](https://twitter.com/ArizePhoenix).
-- 👔 Check out our LinkedIn. TODO(#292): Add link, fix badge
+
+-   🌍 Join our [Slack community](https://join.slack.com/t/arize-ai/shared_invite/zt-1px8dcmlf-fmThhDFD_V_48oU7ALan4Q).
+-   💡 Ask questions and provide feedback in the _#phoenix-support_ channel.
+-   🌟 Leave a star on our [GitHub](https://github.com/Arize-ai/phoenix).
+-   🐞 Report bugs with [GitHub Issues](https://github.com/Arize-ai/phoenix/issues).
+-   🗺️ Check out our [roadmap](https://github.com/orgs/Arize-ai/projects/45) to see where we're heading next.
+-   🎓 Learn the fundamentals of ML observability with our [introductory](https://arize.com/ml-observability-fundamentals/) and [advanced](https://arize.com/blog-course/) courses.
+-   ✏️ Check out our [blog](https://arize.com/blog/). TODO(#291): Add blog filter for Phoenix
+-   ✉️ Subscribe to our mailing list. TODO(#294): Add link
+-   🐦 Follow us on [Twitter](https://twitter.com/ArizePhoenix).
+-   👔 Check out our LinkedIn. TODO(#292): Add link, fix badge
 
 ## Contributing
 
-- 💻 Read our [developer's guide](./DEVELOPMENT.md).
-- 🗣️ Join our [Slack community](https://join.slack.com/t/arize-ai/shared_invite/zt-1px8dcmlf-fmThhDFD_V_48oU7ALan4Q) and chat with us in the *#phoenix-devs* channel.
+-   💻 Read our [developer's guide](./DEVELOPMENT.md).
+-   🗣️ Join our [Slack community](https://join.slack.com/t/arize-ai/shared_invite/zt-1px8dcmlf-fmThhDFD_V_48oU7ALan4Q) and chat with us in the _#phoenix-devs_ channel.
 
 ## License
+
 Arize-Phoenix is licensed under the [Elastic License 2.0 (ELv2)](./LICENSE).

{arize_phoenix-0.0.2rc3 → arize_phoenix-0.0.2rc5}/README.md

@@ -25,12 +25,12 @@ Phoenix provides MLOps insights at lightning speed with zero-config observabilit
 
 **_Phoenix is under active development. APIs may change at any time._**
 
-- [Installation](#installation)
-- [Getting Started](#getting-started)
-- [Documentation](#documentation)
-- [Community](#community)
-- [Contributing](#contributing)
-- [License](#license)
+-   [Installation](#installation)
+-   [Getting Started](#getting-started)
+-   [Documentation](#documentation)
+-   [Community](#community)
+-   [Contributing](#contributing)
+-   [License](#license)
 
 ## Installation
 
@@ -47,8 +47,9 @@ After installing `arize-phoenix` in your Jupyter or Colab environment, open your
 ```python
 import phoenix as px
 
-train_ds, prod_ds = px.load_dataset("sentiment_classification_language_drift")
-px.launch_app(train_ds, prod_ds)
+datasets = px.load_datasets("sentiment_classification_language_drift")
+session = px.launch_app(datasets.primary, datasets.reference)
+session.view()
 ```
 
 Next, visualize your embeddings and inspect problematic clusters of your production data.
@@ -56,6 +57,7 @@ Next, visualize your embeddings and inspect problematic clusters of your product
 TODO(#297): Include GIF where we navigate to embeddings, zoom in and rotate, and select a cluster.
 
 Don't forget to close the app when you're done.
+
 ```
 px.close_app()
 ```
@@ -69,21 +71,23 @@ For in-depth examples and explanations, read the [docs](https://docs.arize.com/p
 ## Community
 
 Join our community to connect with thousands of machine learning practitioners and ML observability enthusiasts.
-- 🌍 Join our [Slack community](https://join.slack.com/t/arize-ai/shared_invite/zt-1px8dcmlf-fmThhDFD_V_48oU7ALan4Q).
-- 💡 Ask questions and provide feedback in the *#phoenix-support* channel.
-- 🌟 Leave a star on our [GitHub](https://github.com/Arize-ai/phoenix).
-- 🐞 Report bugs with [GitHub Issues](https://github.com/Arize-ai/phoenix/issues).
-- 🗺️ Check out our [roadmap](https://github.com/orgs/Arize-ai/projects/45) to see where we're heading next.
-- 🎓 Learn the fundamentals of ML observability with our [introductory](https://arize.com/ml-observability-fundamentals/) and [advanced](https://arize.com/blog-course/) courses.
-- ✏️ Check out our [blog](https://arize.com/blog/). TODO(#291): Add blog filter for Phoenix
-- ✉️ Subscribe to our mailing list. TODO(#294): Add link
-- 🐦 Follow us on [Twitter](https://twitter.com/ArizePhoenix).
-- 👔 Check out our LinkedIn. TODO(#292): Add link, fix badge
+
+-   🌍 Join our [Slack community](https://join.slack.com/t/arize-ai/shared_invite/zt-1px8dcmlf-fmThhDFD_V_48oU7ALan4Q).
+-   💡 Ask questions and provide feedback in the _#phoenix-support_ channel.
+-   🌟 Leave a star on our [GitHub](https://github.com/Arize-ai/phoenix).
+-   🐞 Report bugs with [GitHub Issues](https://github.com/Arize-ai/phoenix/issues).
+-   🗺️ Check out our [roadmap](https://github.com/orgs/Arize-ai/projects/45) to see where we're heading next.
+-   🎓 Learn the fundamentals of ML observability with our [introductory](https://arize.com/ml-observability-fundamentals/) and [advanced](https://arize.com/blog-course/) courses.
+-   ✏️ Check out our [blog](https://arize.com/blog/). TODO(#291): Add blog filter for Phoenix
+-   ✉️ Subscribe to our mailing list. TODO(#294): Add link
+-   🐦 Follow us on [Twitter](https://twitter.com/ArizePhoenix).
+-   👔 Check out our LinkedIn. TODO(#292): Add link, fix badge
 
 ## Contributing
 
-- 💻 Read our [developer's guide](./DEVELOPMENT.md).
-- 🗣️ Join our [Slack community](https://join.slack.com/t/arize-ai/shared_invite/zt-1px8dcmlf-fmThhDFD_V_48oU7ALan4Q) and chat with us in the *#phoenix-devs* channel.
+-   💻 Read our [developer's guide](./DEVELOPMENT.md).
+-   🗣️ Join our [Slack community](https://join.slack.com/t/arize-ai/shared_invite/zt-1px8dcmlf-fmThhDFD_V_48oU7ALan4Q) and chat with us in the _#phoenix-devs_ channel.
 
 ## License
+
 Arize-Phoenix is licensed under the [Elastic License 2.0 (ELv2)](./LICENSE).

arize_phoenix-0.0.2rc5/src/phoenix/__about__.py

@@ -0,0 +1 @@
+__version__ = "0.0.2rc5"

arize_phoenix-0.0.2rc5/src/phoenix/__init__.py

@@ -0,0 +1,2 @@
+from .datasets import Dataset, EmbeddingColumnNames, Schema, load_datasets
+from .session.session import active_session, close_app, launch_app

arize_phoenix-0.0.2rc5/src/phoenix/core/embedding_dimension.py

@@ -0,0 +1,39 @@
+from dataclasses import dataclass
+from typing import Set
+
+from phoenix.datasets.dataset import DatasetType
+from phoenix.datasets.event import EventId
+
+
+@dataclass
+class EmbeddingDimension:
+    name: str
+
+
+def calculate_drift_ratio(events: Set[EventId]) -> float:
+    """
+    Calculates the drift score of the cluster. The score will be a value
+    representing the balance of points between the primary and the reference
+    datasets, and will be on a scale between 1 (all primary) and -1 (all
+    reference), with 0 being an even balance between the two datasets.
+
+    Returns
+    -------
+    drift_ratio : float
+
+    """
+    if not events:
+        return float("nan")
+
+    primary_point_count = 0
+    reference_point_count = 0
+
+    for event in events:
+        if event.dataset_id == DatasetType.PRIMARY:
+            primary_point_count += 1
+        else:
+            reference_point_count += 1
+
+    return (primary_point_count - reference_point_count) / (
+        primary_point_count + reference_point_count
+    )

arize_phoenix-0.0.2rc5/src/phoenix/datasets/__init__.py

@@ -0,0 +1,5 @@
+from .dataset import Dataset
+from .fixtures import load_datasets
+from .schema import EmbeddingColumnNames, Schema
+
+__all__ = ["Dataset", "Schema", "EmbeddingColumnNames", "load_datasets"]

{arize_phoenix-0.0.2rc3 → arize_phoenix-0.0.2rc5}/src/phoenix/datasets/dataset.py

@@ -4,7 +4,7 @@ import sys
 import uuid
 from copy import deepcopy
 from dataclasses import fields, replace
-from datetime import datetime
+from datetime import datetime, timedelta
 from enum import Enum
 from functools import cached_property
 from typing import Any, Dict, List, Optional, Set, Tuple, Union, cast
@@ -37,7 +37,28 @@ if hasattr(sys, "ps1"):
 
 class Dataset:
     """
-    A dataset represents data for a set of inferences. It is represented as a dataframe + schema
+    A dataset to use for analysis using phoenix.
+    Used to construct a phoenix session via px.launch_app
+
+    Parameters
+    ----------
+    dataframe : pandas.DataFrame
+        The pandas dataframe containing the data to analyze
+    schema : phoenix.Schema
+        the schema of the dataset. Maps dataframe columns to the appropriate
+        model inference dimensions (features, predictions, actuals).
+    name : str, optional
+        The name of the dataset. If not provided, a random name will be generated.
+        Is helpful for identifying the dataset in the application.
+
+    Returns
+    -------
+    dataset : Session
+        The session object that can be used to view the application
+
+    Examples
+    --------
+    >>> primary_dataset = px.Dataset(dataframe=production_dataframe, schema=schema, name="primary")
     """
 
     _data_file_name: str = "data.parquet"
@@ -85,9 +106,15 @@ class Dataset:
 
     @cached_property
     def end_time(self) -> datetime:
-        """Returns the datetime of the latest inference in the dataset"""
+        """
+        Returns the datetime of the latest inference in the dataset.
+        end_datetime equals max(timestamp) + 1 microsecond, so that it can be
+        used as part of a right-open interval.
+        """
         timestamp_col_name: str = cast(str, self.schema.timestamp_column_name)
-        end_datetime: datetime = self.__dataframe[timestamp_col_name].max()
+        end_datetime: datetime = self.__dataframe[timestamp_col_name].max() + timedelta(
+            microseconds=1,
+        )  # adding a microsecond, so it can be used as part of a right open interval
         return end_datetime
 
     @property

{arize_phoenix-0.0.2rc3/src/phoenix/server → arize_phoenix-0.0.2rc5/src/phoenix/datasets}/fixtures.py

@@ -1,11 +1,12 @@
 import logging
 import os
 from dataclasses import dataclass, replace
-from typing import Tuple
+from typing import Dict, Tuple
 
 from pandas import read_parquet
 
-from phoenix.datasets import Dataset, EmbeddingColumnNames, Schema
+from .dataset import Dataset
+from .schema import EmbeddingColumnNames, Schema
 
 logger = logging.getLogger(__name__)
 
@@ -189,23 +190,24 @@ FIXTURES: Tuple[Fixture, ...] = (
 NAME_TO_FIXTURE = {fixture.name: fixture for fixture in FIXTURES}
 
 
-def download_fixture_if_missing(fixture_name: str) -> None:
+def download_fixture_if_missing(fixture_name: str) -> Tuple[Dataset, Dataset]:
     """
     Downloads primary and reference datasets for a fixture if they are not found
     locally.
     """
     fixture = _get_fixture_by_name(fixture_name=fixture_name)
     primary_dataset_name, reference_dataset_name = get_dataset_names_from_fixture_name(fixture_name)
-    _download_and_persist_dataset_if_missing(
+    primary_dataset = _download_and_persist_dataset_if_missing(
         dataset_name=primary_dataset_name,
         dataset_url=fixture.primary_dataset_url,
         schema=fixture.primary_schema,
     )
-    _download_and_persist_dataset_if_missing(
+    reference_dataset = _download_and_persist_dataset_if_missing(
         dataset_name=reference_dataset_name,
         dataset_url=fixture.reference_dataset_url,
         schema=fixture.reference_schema,
     )
+    return primary_dataset, reference_dataset
 
 
 def get_dataset_names_from_fixture_name(fixture_name: str) -> Tuple[str, str]:
@@ -223,27 +225,62 @@ def _get_fixture_by_name(fixture_name: str) -> Fixture:
     if the input fixture name does not match any known fixture names.
     """
     if fixture_name not in NAME_TO_FIXTURE:
-        raise ValueError(f'"{fixture_name}" is not a valid fixture name.')
+        valid_fixture_names = ", ".join(NAME_TO_FIXTURE.keys())
+        raise ValueError(f'"{fixture_name}" is invalid. Valid names are: {valid_fixture_names}')
     return NAME_TO_FIXTURE[fixture_name]
 
 
 def _download_and_persist_dataset_if_missing(
     dataset_name: str, dataset_url: str, schema: Schema
-) -> None:
+) -> Dataset:
     """
     Downloads a dataset from the given URL if it is not found locally.
     """
     try:
-        Dataset.from_name(dataset_name)
-        return
+        return Dataset.from_name(dataset_name)
     except FileNotFoundError:
         pass
 
     logger.info(f'Downloading dataset: "{dataset_name}"')
-    Dataset(
+    dataset = Dataset(
         dataframe=read_parquet(dataset_url),
         schema=schema,
         name=dataset_name,
         persist_to_disc=True,
     )
     logger.info("Download complete.")
+    return dataset
+
+
+@dataclass(frozen=True)
+class DatasetDict(Dict[str, Dataset]):
+    """A dictionary of datasets, split out by dataset type (primary, reference)."""
+
+    primary: Dataset
+    reference: Dataset
+
+
+def load_datasets(use_case: str) -> DatasetDict:
+    """
+    Loads the primary and reference datasets for a given use-case.
+
+    Parameters
+    ----------
+        use_case: str
+            Name of the phoenix supported use case
+            Valid values include:
+                - "sentiment_classification_language_drift"
+                - "fashion_mnist"
+                - "ner_token_drift"
+                - "credit_card_fraud"
+                - "click_through_rate"
+
+
+    Returns
+    _______
+        datasets: DatasetDict
+            A dictionary of datasets, split out by dataset type (primary, reference).
+
+    """
+    primary_dataset, reference_dataset = download_fixture_if_missing(use_case)
+    return DatasetDict(primary=primary_dataset, reference=reference_dataset)

{arize_phoenix-0.0.2rc3 → arize_phoenix-0.0.2rc5}/src/phoenix/datasets/validation.py

@@ -69,7 +69,7 @@ def _check_valid_embedding_data(dataframe: DataFrame, schema: Schema) -> List[er
                 embedding_errors.append(
                     err.InvalidEmbeddingVectorDataType(
                         embedding_feature_name=embedding_name,
-                        vector_column_type=str(type(vector_column)),
+                        vector_column_type=str(type(vector)),
                     )
                 )
                 break

{arize_phoenix-0.0.2rc3 → arize_phoenix-0.0.2rc5}/src/phoenix/metrics/metrics.py

@@ -1,13 +1,17 @@
+import math
 import warnings
+from functools import cached_property
 from typing import Union, cast
 
 import numpy as np
 import numpy.typing as npt
 import pandas as pd
 import sklearn  # type: ignore
+from scipy.spatial.distance import euclidean  # type: ignore
 
 from .mixins import (
     BaseMetric,
+    DriftOperator,
     EvaluationMetric,
     OptionalUnaryOperator,
     UnaryOperator,
@@ -18,7 +22,7 @@ from .mixins import (
 
 class Count(OptionalUnaryOperator, ZeroInitialValue, BaseMetric):
     def calc(self, df: pd.DataFrame) -> int:
-        return df.loc[:, self.operand].count() if self.operand else len(df)
+        return df.loc[:, self.operand].count() if self.operand else df.size
 
 
 class Sum(UnaryOperator, BaseMetric):
@@ -29,7 +33,7 @@ class Sum(UnaryOperator, BaseMetric):
 class VectorSum(UnaryOperator, VectorOperator, ZeroInitialValue, BaseMetric):
     def calc(self, df: pd.DataFrame) -> Union[float, npt.NDArray[np.float64]]:
         return np.sum(  # type: ignore
-            df.loc[:, self.operand].to_numpy(),
+            df.loc[:, self.operand].dropna().to_numpy(),
             initial=self.initial_value(),
         )
 
@@ -45,9 +49,7 @@ class VectorMean(UnaryOperator, VectorOperator, BaseMetric):
             warnings.simplefilter("ignore", category=RuntimeWarning)
             return cast(
                 Union[float, npt.NDArray[np.float64]],
-                np.mean(
-                    df.loc[:, self.operand].to_numpy(),
-                ),
+                np.mean(df.loc[:, self.operand].dropna()),
             )
 
 
@@ -80,3 +82,25 @@ class AccuracyScore(EvaluationMetric):
         return cast(
             float, sklearn.metrics.accuracy_score(df.loc[:, self.actual], df.loc[:, self.predicted])
         )
+
+
+class EuclideanDistance(DriftOperator, VectorOperator):
+    @cached_property
+    def ref_value(self) -> Union[float, npt.NDArray[np.float64]]:
+        if self.reference_data is None or self.reference_data.empty:
+            return float("nan")
+        return cast(
+            Union[float, npt.NDArray[np.float64]],
+            np.mean(self.reference_data.loc[:, self.operand].dropna()),
+        )
+
+    def calc(self, df: pd.DataFrame) -> float:
+        if df.empty or (isinstance(self.ref_value, float) and not math.isfinite(self.ref_value)):
+            return float("nan")
+        return cast(
+            float,
+            euclidean(
+                np.mean(df.loc[:, self.operand].dropna()),
+                self.ref_value,
+            ),
+        )

{arize_phoenix-0.0.2rc3 → arize_phoenix-0.0.2rc5}/src/phoenix/metrics/mixins.py

@@ -4,7 +4,7 @@ BaseMetric. Other mixins provide specialized functionalities. Mixins rely
 on cooperative multiple inheritance and method resolution order in Python.
 """
 from abc import ABC, abstractmethod
-from typing import Any, Mapping, Optional, Tuple, Union
+from typing import Any, Mapping, Optional, Tuple
 
 import numpy as np
 import pandas as pd
@@ -21,9 +21,9 @@ class ZeroInitialValue(ABC):
 
 
 class VectorOperator(ABC):
-    shape: Union[int, Tuple[int, ...]]
+    shape: int
 
-    def __init__(self, shape: Union[int, Tuple[int, ...]], **kwargs: Any):
+    def __init__(self, shape: int = 0, **kwargs: Any):
         self.shape = shape
         super().__init__(**kwargs)
 
@@ -89,3 +89,11 @@ class EvaluationMetric(BaseMetric, ABC):
 
     def input_columns(self) -> Tuple[ColumnName, ...]:
         return (self.predicted, self.actual)
+
+
+class DriftOperator(UnaryOperator, BaseMetric, ABC):
+    reference_data: Optional[pd.DataFrame]
+
+    def __init__(self, reference_data: Optional[pd.DataFrame] = None, **kwargs: Any):
+        self.reference_data = reference_data
+        super().__init__(**kwargs)

{arize_phoenix-0.0.2rc3 → arize_phoenix-0.0.2rc5}/src/phoenix/metrics/timeseries.py

@@ -1,7 +1,7 @@
 from datetime import datetime, timedelta
 from functools import partial
 from itertools import accumulate, chain, repeat, takewhile
-from typing import Any, Callable, Generator, Iterable, List, Tuple, Union, cast
+from typing import Any, Callable, Generator, Iterable, List, Tuple, cast
 
 import pandas as pd
 from typing_extensions import TypeAlias
@@ -66,20 +66,20 @@ def _aggregator(
     Calls groupby on the dataframe and apply metric calculations on each group.
     """
     calcs: Tuple[Metric, ...] = tuple(metrics)
-    columns: Union[List[int], slice] = list(
+    columns: List[int] = list(
         set(
             dataframe.columns.get_loc(column_name)
             for calc in calcs
             for column_name in calc.input_columns()
         ),
-    ) or slice(None)
+    )
     return pd.concat(
         chain(
             (pd.DataFrame(),),
             (
                 dataframe.iloc[
                     slice(*row_interval_from_sorted_time_index(dataframe.index, start, end)),
-                    columns,
+                    columns or [0],  # need at least one, so take the first one
                 ]
                 .groupby(group, group_keys=True)
                 .apply(partial(_calculate, calcs=calcs))
@@ -105,16 +105,20 @@ def _groupers(
     """
     Yields pandas.Groupers from time series parameters.
     """
+    if not sampling_interval:
+        return
     divisible = evaluation_window % sampling_interval == timedelta()
-    max_offset = evaluation_window if divisible else end_time - start_time
+    max_offset = end_time - start_time
+    if divisible and evaluation_window < max_offset:
+        max_offset = evaluation_window
     yield from (
         (
-            start_time if divisible else max(start_time, end_time - offset - evaluation_window),
+            (start_time if divisible else end_time - offset) - evaluation_window,
             end_time - offset,
             pd.Grouper(  # type: ignore  # mypy finds the wrong Grouper
                 freq=evaluation_window,
                 origin=end_time,
-                offset=offset,
+                offset=-offset,
                 # Each point in timeseries will be labeled by the end instant of
                 # its evaluation window.
                 label="right",

{arize_phoenix-0.0.2rc3 → arize_phoenix-0.0.2rc5}/src/phoenix/pointcloud/clustering.py

@@ -7,7 +7,7 @@ from hdbscan import HDBSCAN
 from typing_extensions import TypeAlias
 
 RowIndex: TypeAlias = int
-Cluster: TypeAlias = Set[RowIndex]
+RawCluster: TypeAlias = Set[RowIndex]
 Matrix: TypeAlias = npt.NDArray[np.float64]
 
 
@@ -16,9 +16,9 @@ class Hdbscan:
     min_cluster_size: int = 20
     min_samples: float = 1
 
-    def find_clusters(self, mat: Matrix) -> List[Cluster]:
+    def find_clusters(self, mat: Matrix) -> List[RawCluster]:
         cluster_ids: npt.NDArray[np.int_] = HDBSCAN(**asdict(self)).fit_predict(mat)
-        ans: List[Cluster] = [set() for _ in range(np.max(cluster_ids) + 1)]
+        ans: List[RawCluster] = [set() for _ in range(np.max(cluster_ids) + 1)]
         for row_idx, cluster_id in enumerate(cluster_ids):
             if cluster_id > -1:
                 ans[cluster_id].add(row_idx)

{arize_phoenix-0.0.2rc3 → arize_phoenix-0.0.2rc5}/src/phoenix/pointcloud/pointcloud.py

@@ -1,16 +1,17 @@
 from dataclasses import dataclass
-from typing import Dict, Hashable, List, Mapping, Protocol, Set, Tuple, TypeVar
+from typing import Dict, Hashable, List, Mapping, Protocol, Tuple, TypeVar
 
 import numpy as np
 import numpy.typing as npt
 from typing_extensions import TypeAlias
 
-Identifier = TypeVar("Identifier", bound=Hashable)
+from phoenix.pointcloud.clustering import RawCluster
+
 Vector: TypeAlias = npt.NDArray[np.float64]
 Matrix: TypeAlias = npt.NDArray[np.float64]
-ClusterId: TypeAlias = int
 RowIndex: TypeAlias = int
-Cluster: TypeAlias = Set[RowIndex]
+Identifier = TypeVar("Identifier", bound=Hashable)
+ClusterId: TypeAlias = int
 
 
 class DimensionalityReducer(Protocol):
@@ -19,7 +20,7 @@ class DimensionalityReducer(Protocol):
 
 
 class ClustersFinder(Protocol):
-    def find_clusters(self, mat: Matrix) -> List[Cluster]:
+    def find_clusters(self, mat: Matrix) -> List[RawCluster]:
         ...
 
 
@@ -48,15 +49,16 @@ class PointCloud:
         Returns
         -------
         projections : dictionary
-            Projected vectors in the low demension space, mapped back to the
+            Projected vectors in the low dimensional space, mapped back to the
             input vectors' identifiers.
 
-        cluster_membership: dictinary
+        cluster_membership: dictionary
             Cluster membership by way of cluster_ids in the form of integers
             0,1,2,... mapped back to the input vectors' identifiers. Note that
             some vectors may not belong to any cluster and are excluded here.
 
         """
+
         if not data:
             return {}, {}
         identifiers, vectors = zip(*data.items())

{arize_phoenix-0.0.2rc3 → arize_phoenix-0.0.2rc5}/src/phoenix/server/api/input_types/Granularity.py

@@ -40,6 +40,8 @@ class Granularity:
 def to_timestamps(
     time_range: TimeRange, granularity: Granularity
 ) -> Generator[datetime, None, None]:
+    if not granularity.sampling_interval_minutes:
+        return
     yield from (
         takewhile(
             lambda t: time_range.start < t,  # type: ignore