diff --git a/docs/examples/api-reference/operators/join.py b/docs/examples/api-reference/operators/join.py index 8b6438c7..7bb1e880 100644 --- a/docs/examples/api-reference/operators/join.py +++ b/docs/examples/api-reference/operators/join.py @@ -118,7 +118,7 @@ def join_pipeline(cls, tx: Dataset, merchant_category: Dataset): df["timestamp"].tolist() == [datetime(2021, 1, 1, 0, 0, 0, tzinfo=timezone.utc)] * 3 ) - + @mock def test_optional(self, client): # docsnip optional_join @@ -126,6 +126,7 @@ def test_optional(self, client): from fennel.lib import inputs from fennel.connectors import source, Webhook from typing import Optional + webhook = Webhook(name="webhook") @source(webhook.endpoint("Transaction"), disorder="14d", cdc="append") @@ -226,12 +227,19 @@ def join_pipeline(cls, tx: Dataset, merchant_category: Dataset): ), ) import numpy as np + df = client.get_dataset_df("WithCategory") df = df.replace({np.nan: None}) assert df["uid"].tolist() == [1, 1, 2, 3, 3] assert df["merchant"].tolist() == [4, None, 5, 4, 6] assert df["amount"].tolist() == [10, 15, 20, 30, 30] - assert df["category"].tolist() == ["grocery", None, "electronics", "grocery", None] + assert df["category"].tolist() == [ + "grocery", + None, + "electronics", + "grocery", + None, + ] assert ( df["timestamp"].tolist() == [datetime(2021, 1, 1, 0, 0, 0, tzinfo=timezone.utc)] * 5 diff --git a/fennel/datasets/test_schema_validator.py b/fennel/datasets/test_schema_validator.py index f2b4dafa..265f689d 100644 --- a/fennel/datasets/test_schema_validator.py +++ b/fennel/datasets/test_schema_validator.py @@ -2019,7 +2019,7 @@ class ABC: timestamp: datetime @dataset - class XYZJoinedABC: + class XYZJoinedABC: # noqa: F811 user_id: int agent_id: int name: str @@ -2051,7 +2051,7 @@ class ABC: timestamp: datetime @dataset - class XYZJoinedABC: + class XYZJoinedABC: # noqa: F811 user_id: Optional[int] agent_id: int name: str @@ -2082,7 +2082,7 @@ class ABC: timestamp: datetime @dataset - class XYZJoinedABC: + class XYZJoinedABC: # noqa: F811 user_id: Optional[int] agent_id: int name: str @@ -2117,7 +2117,7 @@ class ABC: timestamp: datetime @dataset - class XYZJoinedABC: + class XYZJoinedABC: # noqa: F811 user_id: Optional[int] agent_id: int name: str @@ -2152,7 +2152,7 @@ class ABC: timestamp: datetime @dataset - class XYZJoinedABC: + class XYZJoinedABC: # noqa: F811 user_id: int agent_id: int name: str diff --git a/pyproject.toml b/pyproject.toml index 96e64754..da5d482f 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "fennel-ai" -version = "1.5.40" +version = "1.5.41" description = "The modern realtime feature engineering platform" authors = ["Fennel AI "] packages = [{ include = "fennel" }]