Skip to content
This repository was archived by the owner on Apr 15, 2022. It is now read-only.

Commit a4e57c5

Browse files
author
Epstein
authored
Update management.py (#35)
1 parent dfb446c commit a4e57c5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

splicemachine/ml/management.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727

2828
from ..spark.context import PySpliceContext
2929

30-
SimpleSparkSerializer() # Adds the serializeToBundle function from Mleap
30+
#SimpleSparkSerializer() # Adds the serializeToBundle function from Mleap
3131
CONVERSIONS = PySpliceContext.CONVERSIONS
3232

3333
# For MLeap model deployment
@@ -952,7 +952,7 @@ def _get_mleap_model(self, fittedPipe, df, run_id: str):
952952
:param df: A TRANSFORMED dataframe. ie a dataframe that the pipeline has called .transform() on
953953
:param run_id: (str) the MLFlow run associated with the model
954954
"""
955-
955+
SimpleSparkSerializer() # Adds the serializeToBundle function from Mleap
956956
if 'tmp' not in rbash('ls /').read():
957957
bash('mkdir /tmp')
958958
#Serialize the Spark model into Mleap format

0 commit comments

Comments
 (0)