featurebyte 3.3.0__tar.gz → 3.3.1__tar.gz

{featurebyte-3.3.0 → featurebyte-3.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: featurebyte
-Version: 3.3.0
+Version: 3.3.1
 Summary: Python Library for FeatureOps
 Author: FeatureByte
 Author-email: FeatureByte <it-admin@featurebyte.com>
@@ -149,7 +149,7 @@ Requires-Dist: gevent ; extra == 'server'
 Requires-Dist: google-cloud-bigquery ; extra == 'server'
 Requires-Dist: google-cloud-bigquery-storage ; extra == 'server'
 Requires-Dist: motor ; extra == 'server'
-Requires-Dist: pdfkit ; extra == 'server'
+Requires-Dist: weasyprint ; extra == 'server'
 Requires-Dist: pyhive[hive-pure-sasl] ; extra == 'server'
 Requires-Dist: redis ; extra == 'server'
 Requires-Dist: requests-kerberos ; extra == 'server'

{featurebyte-3.3.0 → featurebyte-3.3.1}/featurebyte/__init__.py

@@ -83,6 +83,7 @@ from featurebyte.models.feature_store import TableStatus
 from featurebyte.models.observation_table import Purpose
 from featurebyte.models.online_store import MySQLOnlineStoreDetails, RedisOnlineStoreDetails
 from featurebyte.models.periodic_task import Crontab
+from featurebyte.models.request_input import DownSamplingInfo, TargetValueSamplingRate
 from featurebyte.models.user_defined_function import FunctionParameter
 from featurebyte.query_graph.model.feature_job_setting import (
     CronFeatureJobSetting,
@@ -94,6 +95,7 @@ from featurebyte.query_graph.model.timestamp_schema import TimestampSchema, Time
 from featurebyte.query_graph.model.window import CalendarWindow
 from featurebyte.query_graph.node.cleaning_operation import (
     AddTimestampSchema,
+    CastToNumeric,
     ColumnCleaningOperation,
     DisguisedValueImputation,
     MissingValueImputation,
@@ -554,6 +556,7 @@ __all__ = [
     "StringValueImputation",
     # other cleaning operation(s)
     "AddTimestampSchema",
+    "CastToNumeric",
     # feature & feature list version specific classes
     "DefaultVersionMode",
     "FeatureVersionInfo",
@@ -566,6 +569,8 @@ __all__ = [
     "TimeInterval",
     "TimeZoneColumn",
     "CalendarWindow",
+    "TargetValueSamplingRate",
+    "DownSamplingInfo",
     # services
     "start",
     "stop",

{featurebyte-3.3.0 → featurebyte-3.3.1}/featurebyte/api/base_table.py

@@ -23,12 +23,13 @@ from featurebyte.api.source_table import AbstractTableData, SourceTable
 from featurebyte.common.doc_util import FBAutoDoc
 from featurebyte.config import Configurations
 from featurebyte.core.mixin import GetAttrMixin, ParentMixin
-from featurebyte.enum import TableDataType, ViewMode
+from featurebyte.enum import DBVarType, TableDataType, ViewMode
 from featurebyte.exception import DuplicatedRecordException, RecordRetrievalException
 from featurebyte.models.base import FeatureByteBaseModel, PydanticObjectId
 from featurebyte.models.feature_store import FeatureStoreModel, TableStatus
 from featurebyte.models.proxy_table import ProxyTableModel
 from featurebyte.query_graph.enum import NodeOutputType, NodeType
+from featurebyte.query_graph.graph import GlobalQueryGraph
 from featurebyte.query_graph.model.column_info import ColumnInfo
 from featurebyte.query_graph.model.common_table import BaseTableData
 from featurebyte.query_graph.model.critical_data_info import CriticalDataInfo
@@ -1081,6 +1082,22 @@ class TableApiObject(
             ]
         return table_data, column_cleaning_operations
 
+    @staticmethod
+    def _prepare_columns_info_for_view(
+        view_node: Node, columns_info: List[ColumnInfo]
+    ) -> List[ColumnInfo]:
+        op_struct = GlobalQueryGraph().extract_operation_structure(node=view_node)
+        col_name_to_dtype = {col_info.name: col_info.dtype for col_info in op_struct.columns}
+        for col_info in columns_info:
+            if (
+                col_info.name in col_name_to_dtype
+                and col_name_to_dtype[col_info.name] not in DBVarType.supported_timestamp_types()
+            ):
+                # casting column with AddTimestampSchema operation may throw error as non-string column
+                # (after casting) should not have timestamp_schema with non-empty format_string
+                col_info.dtype = col_name_to_dtype[col_info.name]
+        return columns_info
+
     @typechecked
     def update_column_entity(self, column_name: str, entity_name: Optional[str]) -> None:
         """

{featurebyte-3.3.0 → featurebyte-3.3.1}/featurebyte/api/deployment.py

@@ -268,6 +268,7 @@ class Deployment(DeletableApiObject):
         columns_rename_mapping: Optional[dict[str, str]] = None,
         point_in_time: Optional[str | datetime] = None,
         use_deployed_tile_tables: bool = False,
+        include_point_in_time: bool = True,
     ) -> BatchFeatureTable:
         """
         Get batch features asynchronously using a batch request table. The batch request features
@@ -293,6 +294,8 @@ class Deployment(DeletableApiObject):
         use_deployed_tile_tables: bool
             Whether to use deployed tile tables for computing the batch feature table. If False,
             tiles will be computed on demand.
+        include_point_in_time: bool
+            Whether to include point in time column in the batch feature table.
 
         Returns
         -------
@@ -330,6 +333,7 @@ class Deployment(DeletableApiObject):
             deployment_id=self.id,
             point_in_time=point_in_time,
             use_deployed_tile_tables=use_deployed_tile_tables,
+            include_point_in_time=include_point_in_time,
         )
         batch_feature_table_doc = self.post_async_task(
             route="/batch_feature_table", payload=payload.json_dict()

{featurebyte-3.3.0 → featurebyte-3.3.1}/featurebyte/api/dimension_table.py

@@ -184,6 +184,9 @@ class DimensionTable(TableApiObject):
             ),
         )
         inserted_graph_node = GlobalQueryGraph().add_node(view_graph_node, input_nodes=[data_node])
+        columns_info = self._prepare_columns_info_for_view(
+            view_node=inserted_graph_node, columns_info=columns_info
+        )
         return DimensionView(
             feature_store=self.feature_store,
             tabular_source=self.tabular_source,

{featurebyte-3.3.0 → featurebyte-3.3.1}/featurebyte/api/event_table.py

@@ -217,6 +217,9 @@ class EventTable(TableApiObject):
             ),
         )
         inserted_graph_node = GlobalQueryGraph().add_node(view_graph_node, input_nodes=[data_node])
+        columns_info = self._prepare_columns_info_for_view(
+            view_node=inserted_graph_node, columns_info=columns_info
+        )
         return EventView(
             feature_store=self.feature_store,
             tabular_source=self.tabular_source,

{featurebyte-3.3.0 → featurebyte-3.3.1}/featurebyte/api/item_table.py

@@ -320,6 +320,9 @@ class ItemTable(TableApiObject):
         inserted_graph_node = GlobalQueryGraph().add_node(
             view_graph_node, input_nodes=[data_node, event_view.node]
         )
+        columns_info = self._prepare_columns_info_for_view(
+            view_node=inserted_graph_node, columns_info=columns_info
+        )
         return ItemView(
             feature_store=self.feature_store,
             tabular_source=self.tabular_source,

{featurebyte-3.3.0 → featurebyte-3.3.1}/featurebyte/api/observation_table.py

@@ -35,6 +35,7 @@ from featurebyte.models.observation_table import (
     Purpose,
     TargetInput,
 )
+from featurebyte.models.request_input import DownSamplingInfo
 from featurebyte.schema.observation_table import (
     ObservationTableCreate,
     ObservationTableListRecord,
@@ -517,6 +518,7 @@ class ObservationTable(PrimaryEntityMixin, MaterializedTableMixin):
         sample_rows: Optional[int] = None,
         sample_from_timestamp: Optional[Union[datetime, str]] = None,
         sample_to_timestamp: Optional[Union[datetime, str]] = None,
+        downsampling_info: Optional[DownSamplingInfo] = None,
     ) -> ObservationTable:
         """
         Creates an ObservationTable from an existing ObservationTable.
@@ -532,6 +534,8 @@ class ObservationTable(PrimaryEntityMixin, MaterializedTableMixin):
             Start of date range to sample from.
         sample_to_timestamp: Optional[Union[datetime, str]]
             End of date range to sample from.
+        downsampling_info: Optional[DownSamplingInfo]
+            Downsampling information to apply when creating the observation table. If provided,
 
         Returns
         -------
@@ -540,9 +544,25 @@ class ObservationTable(PrimaryEntityMixin, MaterializedTableMixin):
         Examples
         --------
         >>> observation_table = catalog.get_observation_table("observation_table")  # doctest: +SKIP
-        >>> sampled_observation_table = source_table.create_observation_table(  # doctest: +SKIP
-        ...     name="<observation_table_name>",
-        ...     sample_rows=desired_sample_size,
+        >>> sampled_observation_table = (
+        ...     observation_table.create_observation_table(  # doctest: +SKIP
+        ...         name="<observation_table_name>",
+        ...         sample_rows=desired_sample_size,
+        ...     )
+        ... )
+
+
+        Create negative downsampled observation table
+
+        >>> neg_downsampled_observation_table = (
+        ...     observation_table.create_observation_table(  # doctest: +SKIP
+        ...         name="<observation_table_name>",
+        ...         downsampling_info=DownSamplingInfo(
+        ...             sampling_rate_per_target_value=[
+        ...                 TargetValueSamplingRate(target_value=0, rate=0.5)
+        ...             ],
+        ...         ),
+        ...     )
         ... )
         """
 
@@ -559,7 +579,10 @@ class ObservationTable(PrimaryEntityMixin, MaterializedTableMixin):
         payload = ObservationTableCreate(
             name=name,
             feature_store_id=self.cached_model.location.feature_store_id,
-            request_input=ObservationTableObservationInput(observation_table_id=self.id),
+            request_input=ObservationTableObservationInput(
+                observation_table_id=self.id,
+                downsampling_info=downsampling_info,
+            ),
             sample_rows=sample_rows,
             sample_from_timestamp=sample_from_timestamp,
             sample_to_timestamp=sample_to_timestamp,

{featurebyte-3.3.0 → featurebyte-3.3.1}/featurebyte/api/relationship.py

@@ -15,7 +15,10 @@ from featurebyte.api.entity import Entity
 from featurebyte.common.doc_util import FBAutoDoc
 from featurebyte.exception import RecordRetrievalException
 from featurebyte.models.base import PydanticObjectId
-from featurebyte.models.relationship import RelationshipInfoModel, RelationshipType
+from featurebyte.models.relationship import (
+    RelationshipInfoModel,
+    RelationshipType,
+)
 from featurebyte.schema.relationship_info import RelationshipInfoUpdate
 
 
@@ -49,6 +52,7 @@ class Relationship(ApiObject):
         "related_entity",
         "relation_table",
         "relation_table_type",
+        "relationship_status",
         "enabled",
         "created_at",
         "updated_at",

{featurebyte-3.3.0 → featurebyte-3.3.1}/featurebyte/api/scd_table.py

@@ -227,6 +227,9 @@ class SCDTable(TableApiObject):
             ),
         )
         inserted_graph_node = GlobalQueryGraph().add_node(view_graph_node, input_nodes=[data_node])
+        columns_info = self._prepare_columns_info_for_view(
+            view_node=inserted_graph_node, columns_info=columns_info
+        )
         return SCDView(
             feature_store=self.feature_store,
             tabular_source=self.tabular_source,
@@ -368,6 +371,9 @@ class SCDTable(TableApiObject):
             ),
         )
         inserted_graph_node = GlobalQueryGraph().add_node(view_graph_node, input_nodes=[data_node])
+        columns_info = self._prepare_columns_info_for_view(
+            view_node=inserted_graph_node, columns_info=columns_info
+        )
         return ChangeView(
             feature_store=self.feature_store,
             tabular_source=self.tabular_source,

{featurebyte-3.3.0 → featurebyte-3.3.1}/featurebyte/api/snapshots_table.py

@@ -170,6 +170,9 @@ class SnapshotsTable(TableApiObject):
             ),
         )
         inserted_graph_node = GlobalQueryGraph().add_node(view_graph_node, input_nodes=[data_node])
+        columns_info = self._prepare_columns_info_for_view(
+            view_node=inserted_graph_node, columns_info=columns_info
+        )
         return SnapshotsView(
             feature_store=self.feature_store,
             tabular_source=self.tabular_source,

{featurebyte-3.3.0 → featurebyte-3.3.1}/featurebyte/api/target.py

@@ -41,6 +41,7 @@ from featurebyte.core.series import Series
 from featurebyte.enum import TargetType
 from featurebyte.models.feature_store import FeatureStoreModel
 from featurebyte.models.target import TargetModel
+from featurebyte.models.target_namespace import PositiveLabelType
 from featurebyte.query_graph.model.common_table import TabularSource
 from featurebyte.schema.target import TargetCreate
 from featurebyte.schema.target_table import TargetTableCreate
@@ -78,6 +79,9 @@ class Target(
     )
     # this is used to store the target_type before the target is saved
     internal_target_type: Optional[TargetType] = Field(default=None, alias="target_type")
+    internal_positive_label: Optional[PositiveLabelType] = Field(
+        default=None, alias="positive_label"
+    )
 
     def _get_create_payload(self) -> dict[str, Any]:
         data = TargetCreate(**self.model_dump(by_alias=True))
@@ -435,6 +439,20 @@ class Target(
             return self.target_namespace.target_type
         return self.internal_target_type
 
+    @property
+    def positive_label(self) -> Optional[PositiveLabelType]:
+        """
+        Positive label
+
+        Returns
+        -------
+        Optional[PositiveLabelType]
+            Positive label
+        """
+        if self.saved:
+            return self.target_namespace.positive_label
+        return self.internal_positive_label
+
     @typechecked
     def update_description(self, description: Optional[str]) -> None:
         """
@@ -488,6 +506,29 @@ class Target(
         else:
             self.internal_target_type = value
 
+    @typechecked
+    def update_positive_label(self, positive_label: PositiveLabelType) -> None:
+        """
+        Update positive label of target.
+
+        The positive label is the value of the positive class in a target for binary classification.
+        Only string, integer and boolean values are supported.
+
+        Parameters
+        ----------
+        positive_label: PositiveLabelType
+            Positive label of the Target for binary classification
+
+        Examples
+        --------
+        >>> target = catalog.get_target("CustomerActive_60days")  # doctest: +SKIP
+        >>> target.update_target_type("CLASSIFICATION")  # doctest: +SKIP
+        """
+        if self.saved:
+            self.target_namespace.update_positive_label(positive_label=positive_label)
+        else:
+            self.internal_positive_label = positive_label
+
     def delete(self) -> None:
         """
         Delete a target from the persistent data store. A target can only be deleted from the persistent

{featurebyte-3.3.0 → featurebyte-3.3.1}/featurebyte/api/target_namespace.py

@@ -19,7 +19,7 @@ from featurebyte.common.doc_util import FBAutoDoc
 from featurebyte.enum import DBVarType, TargetType
 from featurebyte.exception import RecordRetrievalException
 from featurebyte.models.base import PydanticObjectId
-from featurebyte.models.target_namespace import TargetNamespaceModel
+from featurebyte.models.target_namespace import PositiveLabelType, TargetNamespaceModel
 from featurebyte.schema.target_namespace import TargetNamespaceUpdate
 
 
@@ -49,6 +49,9 @@ class TargetNamespace(FeatureOrTargetNamespaceMixin, DeletableApiObject, Savable
     internal_window: Optional[str] = Field(alias="window", default=None)
     internal_dtype: DBVarType = Field(alias="dtype")
     internal_target_type: Optional[TargetType] = Field(alias="target_type", default=None)
+    internal_positive_label: Optional[PositiveLabelType] = Field(
+        alias="positive_label", default=None
+    )
 
     @classmethod
     def create(
@@ -58,6 +61,7 @@ class TargetNamespace(FeatureOrTargetNamespaceMixin, DeletableApiObject, Savable
         dtype: DBVarType,
         window: Optional[str] = None,
         target_type: Optional[TargetType] = None,
+        positive_label: Optional[PositiveLabelType] = None,
     ) -> TargetNamespace:
         """
         Create a new TargetNamespace.
@@ -74,6 +78,10 @@ class TargetNamespace(FeatureOrTargetNamespaceMixin, DeletableApiObject, Savable
             Window of the TargetNamespace
         target_type: Optional[TargetType]
             Type of the Target used to indicate the modeling type of the target
+        positive_label: Optional[PositiveLabelType]
+            Positive label value for classification targets. Only applicable when target_type is
+            CLASSIFICATION. The value type must match the dtype (str for VARCHAR/CHAR, int for INT,
+            bool for BOOL).
 
         Returns
         -------
@@ -89,10 +97,23 @@ class TargetNamespace(FeatureOrTargetNamespaceMixin, DeletableApiObject, Savable
         ...     primary_entity=["customer"],
         ...     target_type=fb.TargetType.REGRESSION,
         ... )
+        >>> classification_target_namespace = fb.TargetNamespace.create(  # doctest: +SKIP
+        ...     name="conversion_target",
+        ...     window="7d",
+        ...     dtype=DBVarType.VARCHAR,
+        ...     primary_entity=["customer"],
+        ...     target_type=fb.TargetType.CLASSIFICATION,
+        ...     positive_label="converted",
+        ... )
         """
         entity_ids = [Entity.get(entity_name).id for entity_name in primary_entity]
         target_namespace = TargetNamespace(
-            name=name, entity_ids=entity_ids, dtype=dtype, window=window, target_type=target_type
+            name=name,
+            entity_ids=entity_ids,
+            dtype=dtype,
+            window=window,
+            target_type=target_type,
+            positive_label=positive_label,
         )
         target_namespace.save()
         return target_namespace
@@ -139,6 +160,20 @@ class TargetNamespace(FeatureOrTargetNamespaceMixin, DeletableApiObject, Savable
         except RecordRetrievalException:
             return self.internal_target_type
 
+    @property
+    def positive_label(self) -> Optional[PositiveLabelType]:
+        """
+        Positive label value for classification targets
+
+        Returns
+        -------
+        Optional[PositiveLabelType]
+        """
+        try:
+            return self.cached_model.positive_label
+        except RecordRetrievalException:
+            return self.internal_positive_label
+
     @property
     def target_ids(self) -> List[PydanticObjectId]:
         """
@@ -205,6 +240,39 @@ class TargetNamespace(FeatureOrTargetNamespaceMixin, DeletableApiObject, Savable
             skip_update_schema_check=True,
         )
 
+    @typechecked
+    def update_positive_label(self, positive_label: PositiveLabelType) -> None:
+        """
+        Update positive label of target.
+
+        The positive label is the value of the positive class in a target for binary classification.
+        Only string, integer and boolean values are supported.
+
+        Parameters
+        ----------
+        positive_label: PositiveLabelType
+            Positive label of the Target for binary classification
+
+        Examples
+        --------
+        >>> target_namespace = fb.TargetNamespace.create(  # doctest: +SKIP
+        ...     name="active_7d_target",
+        ...     window="7d",
+        ...     dtype=DBVarType.BOOL,
+        ...     primary_entity=["customer"],
+        ...     target_type=fb.TargetType.CLASSIFICATION,
+        ... )
+        >>> target_namespace.update_positive_label(True)  # doctest: +SKIP
+        """
+        self.update(
+            update_payload={
+                "positive_label": {"value": positive_label, "observation_table_id": None}
+            },
+            allow_update_local=False,
+            url=f"{self._route}/{self.id}",
+            skip_update_schema_check=True,
+        )
+
     def delete(self) -> None:
         """
         Delete a target namespace from the persistent data store. A target namespace can only be deleted

{featurebyte-3.3.0 → featurebyte-3.3.1}/featurebyte/common/documentation/constants.py

@@ -17,6 +17,7 @@ CONTEXT = "Context"
 CREATE = "Create"
 CREATE_FEATURE = "Create Feature"
 CREATE_FEATURE_GROUP = "Create Feature Group"
+CREATE_OBSERVATION_TABLE = "Create Observation Table"
 CREATE_TABLE = "Create Table"
 CREATE_TARGET = "Create Target"
 CREDENTIAL = "Credential"

{featurebyte-3.3.0 → featurebyte-3.3.1}/featurebyte/common/documentation/custom_nav.py

@@ -18,6 +18,7 @@ from featurebyte.common.documentation.constants import (
     CREATE,
     CREATE_FEATURE,
     CREATE_FEATURE_GROUP,
+    CREATE_OBSERVATION_TABLE,
     CREATE_TABLE,
     CREATE_TARGET,
     CREDENTIAL,
@@ -114,6 +115,7 @@ class BetaWave3Nav(Nav):
         CREATE,
         CREATE_FEATURE,
         CREATE_FEATURE_GROUP,
+        CREATE_OBSERVATION_TABLE,
         CREATE_TABLE,
         CREATE_TARGET,
         CONSTRUCTOR,

{featurebyte-3.3.0 → featurebyte-3.3.1}/featurebyte/common/documentation/documentation_layout.py

@@ -17,6 +17,7 @@ from featurebyte.common.documentation.constants import (
     CREATE,
     CREATE_FEATURE,
     CREATE_FEATURE_GROUP,
+    CREATE_OBSERVATION_TABLE,
     CREATE_TABLE,
     CREATE_TARGET,
     CREDENTIAL,
@@ -856,6 +857,7 @@ def _get_utility_classes_layout() -> List[DocLayoutItem]:
         DocLayoutItem([UTILITY_CLASSES, CLEANING_OPERATION, "ColumnCleaningOperation"]),
         DocLayoutItem([UTILITY_CLASSES, CLEANING_OPERATION, "TableCleaningOperation"]),
         DocLayoutItem([UTILITY_CLASSES, CLEANING_OPERATION, "AddTimestampSchema"]),
+        DocLayoutItem([UTILITY_CLASSES, CLEANING_OPERATION, "CastToNumeric"]),
         DocLayoutItem([UTILITY_CLASSES, ENUMS, "AggFunc"]),
         DocLayoutItem([UTILITY_CLASSES, ENUMS, "TargetType"]),
         DocLayoutItem([UTILITY_CLASSES, ENUMS, "DefaultVersionMode"]),
@@ -904,6 +906,8 @@ def _get_utility_classes_layout() -> List[DocLayoutItem]:
         DocLayoutItem([UTILITY_CLASSES, CREDENTIAL, "OAuthCredential"]),
         DocLayoutItem([UTILITY_CLASSES, CREDENTIAL, "S3StorageCredential"]),
         DocLayoutItem([UTILITY_CLASSES, CREDENTIAL, "UsernamePasswordCredential"]),
+        DocLayoutItem([UTILITY_CLASSES, CREATE_OBSERVATION_TABLE, "TargetValueSamplingRate"]),
+        DocLayoutItem([UTILITY_CLASSES, CREATE_OBSERVATION_TABLE, "DownSamplingInfo"]),
         DocLayoutItem([UTILITY_CLASSES, CREATE_TABLE, "TimestampSchema"]),
         DocLayoutItem([UTILITY_CLASSES, CREATE_TABLE, "TimeZoneColumn"]),
         DocLayoutItem([UTILITY_CLASSES, CREATE_TABLE, "TimeInterval"]),
@@ -1070,6 +1074,7 @@ def _get_observation_table_layout() -> List[DocLayoutItem]:
         DocLayoutItem([OBSERVATION_TABLE, MANAGE, "ObservationTable.update_description"]),
         DocLayoutItem([OBSERVATION_TABLE, MANAGE, "ObservationTable.update_purpose"]),
         DocLayoutItem([OBSERVATION_TABLE, CREATE, "ObservationTable.upload"]),
+        DocLayoutItem([OBSERVATION_TABLE, CREATE, "ObservationTable.create_observation_table"]),
     ]
 
 
@@ -1174,7 +1179,9 @@ def _get_target_layout() -> List[DocLayoutItem]:
         DocLayoutItem([TARGET, SERVE, "Target.compute_target_table"]),
         DocLayoutItem([TARGET, CREATE, "TargetNamespace.create"]),
         DocLayoutItem([TARGET, MANAGE, "TargetNamespace.update_target_type"]),
+        DocLayoutItem([TARGET, MANAGE, "TargetNamespace.update_positive_label"]),
         DocLayoutItem([TARGET, MANAGE, "Target.update_target_type"]),
+        DocLayoutItem([TARGET, MANAGE, "Target.update_positive_label"]),
     ]
 
 

{featurebyte-3.3.0 → featurebyte-3.3.1}/featurebyte/common/utils.py

@@ -286,19 +286,23 @@ def dataframe_to_json(
     }
 
 
-def dataframe_from_json(values: dict[str, Any]) -> pd.DataFrame:
+def apply_type_conversions(
+    dataframe: pd.DataFrame, type_conversions: Optional[dict[Optional[str], DBVarType]]
+) -> pd.DataFrame:
     """
-    Read pandas dataframe from json
+    Apply type conversions to dataframe columns
 
     Parameters
     ----------
-    values: dict[str, Any]
-        Dict containing JSON string and type conversions
+    dataframe: pd.DataFrame
+        Dataframe object
+    type_conversions: Optional[dict[Optional[str], DBVarType]]
+        Conversions to apply on columns
 
     Returns
     -------
     pd.DataFrame
-        Dataframe object
+        Dataframe with applied type conversions
 
     Raises
     ------
@@ -346,8 +350,6 @@ def dataframe_from_json(values: dict[str, Any]) -> pd.DataFrame:
                 pass
         return value
 
-    dataframe = pd.read_json(StringIO(values["data"]), orient="table", convert_dates=False)
-    type_conversions: Optional[dict[Optional[str], DBVarType]] = values.get("type_conversions")
     if type_conversions:
         for col_name, dtype in type_conversions.items():
             # is col_name is None in type_conversions it should apply to the only column in the dataframe
@@ -363,6 +365,25 @@ def dataframe_from_json(values: dict[str, Any]) -> pd.DataFrame:
     return dataframe
 
 
+def dataframe_from_json(values: dict[str, Any]) -> pd.DataFrame:
+    """
+    Read pandas dataframe from json
+
+    Parameters
+    ----------
+    values: dict[str, Any]
+        Dict containing JSON string and type conversions
+
+    Returns
+    -------
+    pd.DataFrame
+        Dataframe object
+    """
+    dataframe = pd.read_json(StringIO(values["data"]), orient="table", convert_dates=False)
+    type_conversions: Optional[dict[Optional[str], DBVarType]] = values.get("type_conversions")
+    return apply_type_conversions(dataframe, type_conversions)
+
+
 def validate_datetime_input(value: Union[datetime, str]) -> str:
     """
     Validate datetime input value

{featurebyte-3.3.0 → featurebyte-3.3.1}/featurebyte/docker/featurebyte.yml

@@ -54,7 +54,7 @@ services:
     hostname: featurebyte-server
     restart: unless-stopped
     container_name: featurebyte-server
-    image: featurebyte/featurebyte-server:3.3.0
+    image: featurebyte/featurebyte-server:3.3.1
     depends_on:
       mongo-rs:
         condition: service_healthy
@@ -89,7 +89,7 @@ services:
     hostname: featurebyte-worker
     restart: unless-stopped
     container_name: featurebyte-worker
-    image: featurebyte/featurebyte-server:3.3.0
+    image: featurebyte/featurebyte-server:3.3.1
     depends_on:
       mongo-rs:
         condition: service_healthy

{featurebyte-3.3.0 → featurebyte-3.3.1}/featurebyte/enum.py

@@ -630,6 +630,7 @@ class InternalName(StrEnum):
     MIGRATION_VERSION = "MIGRATION_VERSION"
     DATAFRAME_ROW_INDEX = "__FB_DATAFRAME_ROW_INDEX"
     TABLE_ROW_INDEX = "__FB_TABLE_ROW_INDEX"
+    TABLE_ROW_WEIGHT = "__FB_TABLE_ROW_WEIGHT"
 
     ONLINE_STORE_RESULT_NAME_COLUMN = "AGGREGATION_RESULT_NAME"
     ONLINE_STORE_VALUE_COLUMN = "VALUE"

{featurebyte-3.3.0 → featurebyte-3.3.1}/featurebyte/exception.py

@@ -495,6 +495,12 @@ class ObservationTableInvalidTargetNameError(BaseUnprocessableEntityError):
     """
 
 
+class ObservationTableInvalidSamplingError(BaseUnprocessableEntityError):
+    """
+    Raise when observation table specifies an invalid sampling configuration
+    """
+
+
 class ObservationTableTargetDefinitionExistsError(BaseUnprocessableEntityError):
     """
     Raise when observation table specifies a target name that already has a definition
@@ -611,3 +617,21 @@ class InvalidComputeOptionValueError(BaseUnprocessableEntityError):
     """
     Raise when an invalid compute option value is provided
     """
+
+
+class InvalidEntityRelationshipError(BaseUnprocessableEntityError):
+    """
+    Raise when an invalid entity relationship is provided
+    """
+
+
+class TargetValidationError(BaseUnprocessableEntityError):
+    """
+    Raise when target validation fails
+    """
+
+
+class DeploymentSqlGenerationError(BaseUnprocessableEntityError):
+    """
+    Raise when there is an error during deployment SQL generation
+    """

{featurebyte-3.3.0 → featurebyte-3.3.1}/featurebyte/models/batch_feature_table.py

@@ -4,6 +4,7 @@ BatchFeatureTable related model(s)
 
 from __future__ import annotations
 
+from datetime import datetime
 from typing import List, Optional
 
 import pymongo
@@ -27,6 +28,7 @@ class BatchFeatureTableModel(MaterializedTableModel):
     request_input: Optional[BatchRequestInput] = Field(default=None)
     deployment_id: PydanticObjectId
     parent_batch_feature_table_name: Optional[str] = Field(default=None)
+    point_in_time: Optional[datetime] = Field(default=None)
 
     class Settings(MaterializedTableModel.Settings):
         """