Skip to content

Commit

Permalink
adding docstring for output vars
Browse files Browse the repository at this point in the history
  • Loading branch information
namera9 committed Apr 4, 2024
1 parent d896f8f commit ff4db63
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions src/sasctl/pzmm/import_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(...,
Expand All @@ -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:
Expand Down

0 comments on commit ff4db63

Please sign in to comment.