From 83b4ff3bb3c93cd064c5f18f52eb23bbbd530a73 Mon Sep 17 00:00:00 2001 From: Alexandru Ormenisan Date: Wed, 30 Oct 2024 09:39:22 +0100 Subject: [PATCH] Model provenance - argument description --- python/hsml/llm/signature.py | 2 ++ python/hsml/python/signature.py | 2 ++ python/hsml/sklearn/signature.py | 2 ++ python/hsml/tensorflow/signature.py | 2 ++ python/hsml/torch/signature.py | 2 ++ 5 files changed, 10 insertions(+) diff --git a/python/hsml/llm/signature.py b/python/hsml/llm/signature.py index 9ac7db9ff..05ff003eb 100644 --- a/python/hsml/llm/signature.py +++ b/python/hsml/llm/signature.py @@ -56,6 +56,8 @@ def create_model( `""`. input_example: Optionally an input example that represents a single input for the model, defaults to `None`. model_schema: Optionally a model schema for the model inputs and/or outputs. + feature_view: Optionally a feature view object returned by querying the feature store. If the feature view is not provided, the model will not have access to provenance. + training_dataset_version: Optionally a training dataset version. If training dataset version is not provided, but the feature view is provided, the training dataset version used will be the last accessed training dataset of the feature view, within the code/notebook that reads the feature view and training dataset and then creates the model. # Returns `Model`. The model metadata object. diff --git a/python/hsml/python/signature.py b/python/hsml/python/signature.py index 1bb5fa8f7..fa704aaab 100644 --- a/python/hsml/python/signature.py +++ b/python/hsml/python/signature.py @@ -56,6 +56,8 @@ def create_model( `""`. input_example: Optionally an input example that represents a single input for the model, defaults to `None`. model_schema: Optionally a model schema for the model inputs and/or outputs. + feature_view: Optionally a feature view object returned by querying the feature store. If the feature view is not provided, the model will not have access to provenance. + training_dataset_version: Optionally a training dataset version. If training dataset version is not provided, but the feature view is provided, the training dataset version used will be the last accessed training dataset of the feature view, within the code/notebook that reads the feature view and training dataset and then creates the model. # Returns `Model`. The model metadata object. diff --git a/python/hsml/sklearn/signature.py b/python/hsml/sklearn/signature.py index f8816febb..4c145a96a 100644 --- a/python/hsml/sklearn/signature.py +++ b/python/hsml/sklearn/signature.py @@ -56,6 +56,8 @@ def create_model( `""`. input_example: Optionally an input example that represents a single input for the model, defaults to `None`. model_schema: Optionally a model schema for the model inputs and/or outputs. + feature_view: Optionally a feature view object returned by querying the feature store. If the feature view is not provided, the model will not have access to provenance. + training_dataset_version: Optionally a training dataset version. If training dataset version is not provided, but the feature view is provided, the training dataset version used will be the last accessed training dataset of the feature view, within the code/notebook that reads the feature view and training dataset and then creates the model. # Returns `Model`. The model metadata object. diff --git a/python/hsml/tensorflow/signature.py b/python/hsml/tensorflow/signature.py index 1f83c5496..e24d20e65 100644 --- a/python/hsml/tensorflow/signature.py +++ b/python/hsml/tensorflow/signature.py @@ -56,6 +56,8 @@ def create_model( `""`. input_example: Optionally an input example that represents a single input for the model, defaults to `None`. model_schema: Optionally a model schema for the model inputs and/or outputs. + feature_view: Optionally a feature view object returned by querying the feature store. If the feature view is not provided, the model will not have access to provenance. + training_dataset_version: Optionally a training dataset version. If training dataset version is not provided, but the feature view is provided, the training dataset version used will be the last accessed training dataset of the feature view, within the code/notebook that reads the feature view and training dataset and then creates the model. # Returns `Model`. The model metadata object. diff --git a/python/hsml/torch/signature.py b/python/hsml/torch/signature.py index 5234d110a..bab488974 100644 --- a/python/hsml/torch/signature.py +++ b/python/hsml/torch/signature.py @@ -56,6 +56,8 @@ def create_model( `""`. input_example: Optionally an input example that represents a single input for the model, defaults to `None`. model_schema: Optionally a model schema for the model inputs and/or outputs. + feature_view: Optionally a feature view object returned by querying the feature store. If the feature view is not provided, the model will not have access to provenance. + training_dataset_version: Optionally a training dataset version. If training dataset version is not provided, but the feature view is provided, the training dataset version used will be the last accessed training dataset of the feature view, within the code/notebook that reads the feature view and training dataset and then creates the model. # Returns `Model`. The model metadata object.