Skip to content

Commit b5609e3

Browse files
committed
Fix incorrect pickleType default value in writeScoreCode.
1 parent 9fd70ee commit b5609e3

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/sasctl/pzmm/writeScoreCode.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ def writeScoreCode(
2828
scoreCAS=True,
2929
isBinaryModel=False,
3030
binaryString=None,
31-
pickleType=None,
31+
pickleType='pickle',
3232
):
3333
"""
3434
Writes a Python score code file based on training data used to generate the model
@@ -110,8 +110,8 @@ def writeScoreCode(
110110
Sets whether the H2O model provided is a binary model or a MOJO model. By default False.
111111
binaryString : string, optional
112112
Binary string representation of the model object. By default None.
113-
pickleType : string optional
114-
Indicator for MLFlow models, which may pickle by non-standard methods. By default None.
113+
pickleType : string, optional
114+
Indicator for MLFlow models, which may pickle by non-standard methods. By default 'pickle'.
115115
"""
116116
# Check if binary string model
117117
if binaryString is not None:

0 commit comments

Comments
 (0)