@@ -295,7 +295,7 @@ def score(var1, var2, var3, var4):
295
295
296
296
if preprocess_function :
297
297
self ._add_preprocess_code (preprocess_function )
298
-
298
+
299
299
# SAS Viya 3.5 model
300
300
if model_id :
301
301
mas_code , cas_code = self ._viya35_score_code_import (
@@ -2260,13 +2260,10 @@ def _viya35_score_code_import(
2260
2260
mr .update_model (model )
2261
2261
return mas_code , cas_code
2262
2262
2263
- def _add_preprocess_code (
2264
- self ,
2265
- preprocess_function : Callable [DataFrame , DataFrame ]
2266
- ):
2263
+ def _add_preprocess_code (self , preprocess_function : Callable [DataFrame , DataFrame ]):
2267
2264
"""
2268
2265
Places the given preprocess function, which must both take a DataFrame as an argument
2269
- and return a DataFrame, into the score code. If the preprocess function does not
2266
+ and return a DataFrame, into the score code. If the preprocess function does not
2270
2267
return anything, an error is thrown.
2271
2268
2272
2269
Parameters
@@ -2275,14 +2272,15 @@ def _add_preprocess_code(
2275
2272
The preprocess function to be added to the score code.
2276
2273
"""
2277
2274
import inspect
2275
+
2278
2276
preprocess_code = inspect .getsource (preprocess_function )
2279
2277
if not "return" in preprocess_code :
2280
2278
raise ValueError (
2281
- "The given score code does not return a value. " +
2282
- "To allow for the score code to work correctly, please ensure the preprocessed " +
2283
- "data is returned."
2279
+ "The given score code does not return a value. "
2280
+ + "To allow for the score code to work correctly, please ensure the preprocessed "
2281
+ + "data is returned."
2284
2282
)
2285
- if self .score_code [- 1 ] == ' \n ' :
2283
+ if self .score_code [- 1 ] == " \n " :
2286
2284
self .score_code += preprocess_code
2287
2285
else :
2288
- self .score_code += ' \n ' + preprocess_code
2286
+ self .score_code += " \n " + preprocess_code
0 commit comments