Fix?

Signed-off-by: Kevin Zhang <kzhang@tecton.ai>
feast-dev · feast-ci-bot · Aug 19, 2022 · Aug 5, 2022 · Aug 10, 2022 · Aug 10, 2022
commit 4a969e77b65242767b78bea4fea49f5976a64cbd
diff --git a/README.md b/README.md
@@ -152,7 +152,7 @@ The list below contains the functionality that contributors are planning to deve
   * [x] [Redshift source](https://docs.feast.dev/reference/data-sources/redshift)
   * [x] [BigQuery source](https://docs.feast.dev/reference/data-sources/bigquery)
   * [x] [Parquet file source](https://docs.feast.dev/reference/data-sources/file)
-  * [x] [Synapse source (community plugin)](https://github.com/Azure/feast-azure)
+  * [x] [Synapse source (contrib plugin)](https://docs.feast.dev/reference/data-sources/mssql)
   * [x] [Hive (community plugin)](https://github.com/baineng/feast-hive)
   * [x] [Postgres (contrib plugin)](https://docs.feast.dev/reference/data-sources/postgres)
   * [x] [Spark (contrib plugin)](https://docs.feast.dev/reference/data-sources/spark)
@@ -161,7 +161,7 @@ The list below contains the functionality that contributors are planning to deve
   * [x] [Snowflake](https://docs.feast.dev/reference/offline-stores/snowflake)
   * [x] [Redshift](https://docs.feast.dev/reference/offline-stores/redshift)
   * [x] [BigQuery](https://docs.feast.dev/reference/offline-stores/bigquery)
-  * [x] [Synapse (community plugin)](https://github.com/Azure/feast-azure)
+  * [x] [Synapse (contrib plugin)](https://docs.feast.dev/reference/offline-stores/mssql.md)
   * [x] [Hive (community plugin)](https://github.com/baineng/feast-hive)
   * [x] [Postgres (contrib plugin)](https://docs.feast.dev/reference/offline-stores/postgres)
   * [x] [Trino (contrib plugin)](https://github.com/Shopify/feast-trino)

@@ -71,6 +71,7 @@
   * [Spark (contrib)](reference/data-sources/spark.md)
   * [PostgreSQL (contrib)](reference/data-sources/postgres.md)
   * [Trino (contrib)](reference/data-sources/trino.md)
+  * [Synapse/MsSql (contrib)](reference/data-sources/mssql.md)
 * [Offline stores](reference/offline-stores/README.md)
   * [Overview](reference/offline-stores/overview.md)
   * [File](reference/offline-stores/file.md)
@@ -80,6 +81,7 @@
   * [Spark (contrib)](reference/offline-stores/spark.md)
   * [PostgreSQL (contrib)](reference/offline-stores/postgres.md)
   * [Trino (contrib)](reference/offline-stores/trino.md)
+  * [Synapse/MsSql (contrib)](reference/offline-stores/mssql.md)
 * [Online stores](reference/online-stores/README.md)
   * [SQLite](reference/online-stores/sqlite.md)
   * [Snowflake](reference/online-stores/snowflake.md)
@@ -91,6 +93,7 @@
   * [Local](reference/providers/local.md)
   * [Google Cloud Platform](reference/providers/google-cloud-platform.md)
   * [Amazon Web Services](reference/providers/amazon-web-services.md)
+  * [Azure](reference/providers/azure.md)
 * [Feature repository](reference/feature-repository/README.md)
   * [feature\_store.yaml](reference/feature-repository/feature-store-yaml.md)
   * [.feastignore](reference/feature-repository/feast-ignore.md)

@@ -10,7 +10,7 @@ The list below contains the functionality that contributors are planning to deve
   * [x] [Redshift source](https://docs.feast.dev/reference/data-sources/redshift)
   * [x] [BigQuery source](https://docs.feast.dev/reference/data-sources/bigquery)
   * [x] [Parquet file source](https://docs.feast.dev/reference/data-sources/file)
-  * [x] [Synapse source (community plugin)](https://github.com/Azure/feast-azure)
+  * [x] [Synapse source (contrib plugin)](https://docs.feast.dev/reference/data-sources/mssql)
   * [x] [Hive (community plugin)](https://github.com/baineng/feast-hive)
   * [x] [Postgres (contrib plugin)](https://docs.feast.dev/reference/data-sources/postgres)
   * [x] [Spark (contrib plugin)](https://docs.feast.dev/reference/data-sources/spark)
@@ -19,7 +19,7 @@ The list below contains the functionality that contributors are planning to deve
   * [x] [Snowflake](https://docs.feast.dev/reference/offline-stores/snowflake)
   * [x] [Redshift](https://docs.feast.dev/reference/offline-stores/redshift)
   * [x] [BigQuery](https://docs.feast.dev/reference/offline-stores/bigquery)
-  * [x] [Synapse (community plugin)](https://github.com/Azure/feast-azure)
+  * [x] [Synapse (contrib plugin)](https://docs.feast.dev/reference/offline-stores/mssql.md)
   * [x] [Hive (community plugin)](https://github.com/baineng/feast-hive)
   * [x] [Postgres (contrib plugin)](https://docs.feast.dev/reference/offline-stores/postgres)
   * [x] [Trino (contrib plugin)](https://github.com/Shopify/feast-trino)

@@ -1,6 +1,6 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT license.
-
+import warnings
 from datetime import datetime, timedelta
 from typing import Any, Callable, Dict, List, Optional, Sequence, Tuple, Union
 
@@ -28,11 +28,19 @@
     make_tzaware,
 )
 
+# Make sure spark warning doesn't raise more than once.
+warnings.simplefilter("once", RuntimeWarning)
+
 DEFAULT_BATCH_SIZE = 10_000
 
 
 class AzureProvider(Provider):
     def __init__(self, config: RepoConfig):
+        warnings.warn(
+            "The azure provider  is an experimental feature in alpha development. "
+            "Some functionality may still be unstable so functionality can change in the future.",
+            RuntimeWarning,
+        )
         self.repo_config = config
         self.offline_store = get_offline_store_from_config(config.offline_store)
         self.online_store = (
@@ -69,6 +77,11 @@ def teardown_infra(
         tables: Sequence[FeatureView],
         entities: Sequence[Entity],
     ) -> None:
+        warnings.warn(
+            "The azure provider  is an experimental feature in alpha development. "
+            "Some functionality may still be unstable so functionality can change in the future.",
+            RuntimeWarning,
+        )
         if self.online_store:
             self.online_store.teardown(self.repo_config, tables, entities)
 
@@ -93,6 +106,7 @@ def online_read(
         entity_keys: List[EntityKeyProto],
         requested_features: List[str] = None,
     ) -> List[Tuple[Optional[datetime], Optional[Dict[str, ValueProto]]]]:
+
         result = []
         if self.online_store:
             result = self.online_store.online_read(

@@ -1,6 +1,6 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT license.
-
+import warnings
 from datetime import datetime
 from typing import Any, Dict, List, Optional, Set, Tuple, Union
 
@@ -33,6 +33,9 @@
 from feast.saved_dataset import SavedDatasetStorage
 from feast.usage import log_exceptions_and_usage
 
+# Make sure spark warning doesn't raise more than once.
+warnings.simplefilter("once", RuntimeWarning)
+
 EntitySchema = Dict[str, np.dtype]
 
 
@@ -64,6 +67,11 @@ def pull_latest_from_table_or_query(
         start_date: datetime,
         end_date: datetime,
     ) -> RetrievalJob:
+        warnings.warn(
+            "The synapse/mssql offline store is an experimental feature in alpha development. "
+            "Some functionality may still be unstable so functionality can change in the future.",
+            RuntimeWarning,
+        )
         assert type(data_source).__name__ == "MsSqlServerSource"
         from_expression = data_source.get_table_query_string().replace("`", "")
 
@@ -110,6 +118,11 @@ def pull_all_from_table_or_query(
         end_date: datetime,
     ) -> RetrievalJob:
         assert type(data_source).__name__ == "MsSqlServerSource"
+        warnings.warn(
+            "The synapse/mssql offline store is an experimental feature in alpha development. "
+            "Some functionality may still be unstable so functionality can change in the future.",
+            RuntimeWarning,
+        )
         from_expression = data_source.get_table_query_string().replace("`", "")
         timestamps = [timestamp_field]
         field_string = ", ".join(join_key_columns + feature_name_columns + timestamps)
@@ -143,6 +156,11 @@ def get_historical_features(
         project: str,
         full_feature_names: bool = False,
     ) -> RetrievalJob:
+        warnings.warn(
+            "The synapse/mssql offline store is an experimental feature in alpha development. "
+            "Some functionality may still be unstable so functionality can change in the future.",
+            RuntimeWarning,
+        )
         expected_join_keys = _get_join_keys(project, feature_views, registry)
 
         assert isinstance(config.offline_store, MsSqlServerOfflineStoreConfig)

@@ -1,6 +1,6 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT license.
-
+import warnings
 import json
 from typing import Callable, Dict, Iterable, Optional, Tuple
 
@@ -16,6 +16,8 @@
 from feast.repo_config import RepoConfig
 from feast.value_type import ValueType
 
+# Make sure spark warning doesn't raise more than once.
+warnings.simplefilter("once", RuntimeWarning)
 
 class MsSqlServerOptions:
     """
@@ -111,6 +113,11 @@ def __init__(
         tags: Optional[Dict[str, str]] = None,
         owner: Optional[str] = None,
     ):
+        warnings.warn(
+            "The synapse/mssql data source is an experimental feature in alpha development. "
+            "Some functionality may still be unstable so functionality can change in the future.",
+            RuntimeWarning,
+        )
         self._mssqlserver_options = MsSqlServerOptions(
             connection_str=connection_str, table_ref=table_ref
         )