From ff4db6331b88cffc7816481ebb5cc53934e2505e Mon Sep 17 00:00:00 2001 From: Nabeel Merali Date: Thu, 4 Apr 2024 09:48:45 -0400 Subject: [PATCH] adding docstring for output vars --- src/sasctl/pzmm/import_model.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/sasctl/pzmm/import_model.py b/src/sasctl/pzmm/import_model.py index 7ec9d658..88d9eade 100644 --- a/src/sasctl/pzmm/import_model.py +++ b/src/sasctl/pzmm/import_model.py @@ -287,6 +287,9 @@ def import_model( index should match the index of the target outcome in target_values. If target_values are not given, this index should indicate whether the the target probability variable is the first or second variable returned by the model. The default value is 1. + score_wrapper : bool, optional + if the score wrapper method is being used, it shouldn't write the score code as the score wrapper function will handle + the formulation of the score code. kwargs : dict, optional Other keyword arguments are passed to the following function: * sasctl.pzmm.ScoreCode.write_score_code(..., @@ -306,6 +309,7 @@ def import_model( model_files : dict, str, or Path Dictionary representation of all files or the path the model files were generated from. + :param score_wrapper: """ # For mlflow models, overwrite the provided or default pickle_type if mlflow_details: