databricks-feature-engineering 0.8.0__tar.gz

databricks_feature_engineering-0.8.0/LICENSE.md

@@ -0,0 +1,31 @@
+**DB license**
+
+Copyright (2022) Databricks, Inc.
+
+This library (the "Software") may not be used except in connection with the Licensee's use of the Databricks Platform Services
+pursuant to an Agreement (defined below) between Licensee (defined below) and Databricks, Inc. ("Databricks"). This Software
+shall be deemed part of the Downloadable Services under the Agreement, or if the Agreement does not define Downloadable Services,
+Subscription Services, or if neither are defined then the term in such Agreement that refers to the applicable Databricks Platform
+Services (as defined below) shall be substituted herein for "Downloadable Services". Licensee's use of the Software must comply at
+all times with any restrictions applicable to the Downlodable Services and Subscription Services, generally, and must be used in
+accordance with any applicable documentation.
+
+Additionally, and notwithstanding anything in the Agreement to the contrary:
+* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
+OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR
+IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+If you have not agreed to an Agreement or otherwise do not agree to these terms, you may not use the Software.
+
+This license terminates automatically upon the termination of the Agreement or Licensee's breach of these terms.
+
+Agreement: the agreement between Databricks and Licensee governing the use of the Databricks Platform Services, which shall be, with
+respect to Databricks, the Databricks Terms of Service located at www.databricks.com/termsofservice, and with respect to Databricks
+Community Edition, the Community Edition Terms of Service located at www.databricks.com/ce-termsofuse, in each case unless Licensee
+has entered into a separate written agreement with Databricks governing the use of the applicable Databricks Platform Services.
+
+Databricks Platform Services: the Databricks services or the Databricks Community Edition services, according to where the Software is used.
+
+Licensee: the user of the Software, or, if the Software is being used on behalf of a company, the company.
+

databricks_feature_engineering-0.8.0/NOTICE.md

@@ -0,0 +1,16 @@
+## Databricks Feature Engineering Client
+
+Copyright (2023) Databricks, Inc.
+
+This Software includes software developed at Databricks (https://www.databricks.com/) and its use is subject
+to the included LICENSE.md file.
+
+Additionally, this Software contains code from the following open source projects:
+
+* azure-cosmos - MIT License
+* boto3 - Apache 2 (https://www.apache.org/licenses/LICENSE-2.0)
+* MLflow - Apache 2
+* PyYAML - MIT License
+* SQLAlchemy - MIT License
+* PyMysql - MIT License
+* PyODBC - MIT License

databricks_feature_engineering-0.8.0/PKG-INFO

@@ -0,0 +1,75 @@
+Metadata-Version: 2.1
+Name: databricks-feature-engineering
+Version: 0.8.0
+Summary: Databricks Feature Engineering Client
+Author-email: Databricks <feedback@databricks.com>
+License: Databricks Proprietary License
+Keywords: databricks,feature engineering,feature store
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.8.10
+Description-Content-Type: text/markdown
+License-File: LICENSE.md
+License-File: NOTICE.md
+Requires-Dist: mlflow-skinny[databricks]<3,>=2.11.0
+Requires-Dist: pyyaml<7,>=6
+Requires-Dist: boto3<2,>=1.16.7
+Requires-Dist: dbl-tempo<1,>=0.1.26
+Requires-Dist: azure-cosmos==4.3.1
+Requires-Dist: numpy<2,>=1.19.2
+Requires-Dist: protobuf<5,>=3.12.0
+Requires-Dist: flask<3,>=1.1.2
+Requires-Dist: sqlparse<1,>=0.4.0
+
+The Databricks Feature Engineering client is used to:
+* Create, read, and write feature tables
+* Train models on feature data
+* Publish feature tables to online stores for real-time serving
+* Upgrade workspace feature table metadata to Unity Catalog
+
+# Documentation
+
+Documentation can be found per-cloud at:
+- [AWS](https://docs.databricks.com/applications/machine-learning/feature-store/index.html)
+- [Azure](https://docs.microsoft.com/en-us/azure/databricks/applications/machine-learning/feature-store/)
+- [GCP](https://docs.gcp.databricks.com/applications/machine-learning/feature-store/index.html)
+
+For release notes, see
+- [AWS](https://docs.databricks.com/release-notes/feature-store/databricks-feature-store.html)
+- [Azure](https://docs.microsoft.com/en-us/azure/databricks/release-notes/feature-store/databricks-feature-store)
+- [GCP](https://docs.gcp.databricks.com/release-notes/feature-store/databricks-feature-store.html)
+
+# Limitations
+
+This library can only run on Databricks runtimes. It can be installed locally or in CI/CD environments for unit testing.
+
+**DB license**
+
+Copyright (2022) Databricks, Inc.
+
+This library (the "Software") may not be used except in connection with the Licensee's use of the Databricks Platform Services
+pursuant to an Agreement (defined below) between Licensee (defined below) and Databricks, Inc. ("Databricks"). This Software
+shall be deemed part of the Downloadable Services under the Agreement, or if the Agreement does not define Downloadable Services,
+Subscription Services, or if neither are defined then the term in such Agreement that refers to the applicable Databricks Platform
+Services (as defined below) shall be substituted herein for "Downloadable Services". Licensee's use of the Software must comply at
+all times with any restrictions applicable to the Downlodable Services and Subscription Services, generally, and must be used in
+accordance with any applicable documentation.
+
+Additionally, and notwithstanding anything in the Agreement to the contrary:
+* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
+OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR
+IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+If you have not agreed to an Agreement or otherwise do not agree to these terms, you may not use the Software.
+
+This license terminates automatically upon the termination of the Agreement or Licensee's breach of these terms.
+
+Agreement: the agreement between Databricks and Licensee governing the use of the Databricks Platform Services, which shall be, with
+respect to Databricks, the Databricks Terms of Service located at www.databricks.com/termsofservice, and with respect to Databricks
+Community Edition, the Community Edition Terms of Service located at www.databricks.com/ce-termsofuse, in each case unless Licensee
+has entered into a separate written agreement with Databricks governing the use of the applicable Databricks Platform Services.
+
+Databricks Platform Services: the Databricks services or the Databricks Community Edition services, according to where the Software is used.
+
+Licensee: the user of the Software, or, if the Software is being used on behalf of a company, the company.
+

databricks_feature_engineering-0.8.0/README.md

@@ -0,0 +1,21 @@
+The Databricks Feature Engineering client is used to:
+* Create, read, and write feature tables
+* Train models on feature data
+* Publish feature tables to online stores for real-time serving
+* Upgrade workspace feature table metadata to Unity Catalog
+
+# Documentation
+
+Documentation can be found per-cloud at:
+- [AWS](https://docs.databricks.com/applications/machine-learning/feature-store/index.html)
+- [Azure](https://docs.microsoft.com/en-us/azure/databricks/applications/machine-learning/feature-store/)
+- [GCP](https://docs.gcp.databricks.com/applications/machine-learning/feature-store/index.html)
+
+For release notes, see
+- [AWS](https://docs.databricks.com/release-notes/feature-store/databricks-feature-store.html)
+- [Azure](https://docs.microsoft.com/en-us/azure/databricks/release-notes/feature-store/databricks-feature-store)
+- [GCP](https://docs.gcp.databricks.com/release-notes/feature-store/databricks-feature-store.html)
+
+# Limitations
+
+This library can only run on Databricks runtimes. It can be installed locally or in CI/CD environments for unit testing.

databricks_feature_engineering-0.8.0/databricks/__init__.py

@@ -0,0 +1,2 @@
+# https://packaging.python.org/guides/packaging-namespace-packages/#pkgutil-style-namespace-packages
+__path__ = __import__("pkgutil").extend_path(__path__, __name__)

databricks_feature_engineering-0.8.0/databricks/_feature_store_pkg_metadata/__init__.py

@@ -0,0 +1,2 @@
+# https://packaging.python.org/guides/packaging-namespace-packages/#pkgutil-style-namespace-packages
+__path__ = __import__("pkgutil").extend_path(__path__, __name__)

databricks_feature_engineering-0.8.0/databricks/feature_engineering/__init__.py

@@ -0,0 +1,59 @@
+import importlib.metadata
+
+
+def is_package_installed(package_name):
+    try:
+        # Attempt to get the distribution of the specified package
+        importlib.metadata.version(package_name)
+        return True
+    except Exception:
+        return False
+
+
+# When one of the proto files (e.g feature_catalog_pb2) is imported from both databricks-feature-store and databricks-feature-engineering
+# there will be a type error since the proto file names are the same but contents are slightly different.
+# This function is a best effort to catch and surface a better error message.
+def duplicate_proto_detection():
+    try:
+        from databricks.ml_features.api.proto import feature_catalog_pb2
+    except TypeError as e:
+        if "duplicate file name" in str(e) and is_package_installed(
+            "databricks-feature-store"
+        ):
+            print(
+                "An error can occur when importing from both databricks-feature-store "
+                "and databricks-feature-engineering. Run '%pip uninstall databricks-feature-store' to uninstall databricks-feature-store, then "
+                "run 'dbutils.library.restartPython()' to restart the Python interpreter, and try again. If you would like to use the specific databricks-feature-store version, "
+                "use the corresponding Databricks Runtime for Machine Learning version (see https://docs.databricks.com/release-notes/runtime/index.html#feature-engineering-compatibility-matrix) instead."
+            )
+        raise e
+
+
+duplicate_proto_detection()
+
+# Inject FeatureEngineeringClient specific things
+from databricks.ml_features.utils.logging_utils import _configure_feature_store_loggers
+
+_configure_feature_store_loggers(root_module_name=__name__)
+
+# Support sugar-syntax `from databricks.feature_engineering import FeatureEngineeringClient`, etc.
+from databricks.feature_engineering.client import FeatureEngineeringClient
+from databricks.feature_engineering.entities.aggregation import Aggregation, Window
+from databricks.feature_engineering.entities.cron_schedule import CronSchedule
+from databricks.feature_engineering.entities.feature_aggregations import (
+    FeatureAggregations,
+)
+from databricks.feature_engineering.entities.feature_function import FeatureFunction
+from databricks.feature_engineering.entities.feature_lookup import FeatureLookup
+from databricks.feature_engineering.upgrade_client import UpgradeClient
+
+__all__ = [
+    "FeatureEngineeringClient",
+    "Aggregation",
+    "Window",
+    "CronSchedule",
+    "FeatureAggregations",
+    "FeatureLookup",
+    "FeatureFunction",
+    "UpgradeClient",
+]