Skip to content

Commit e5938a5

Browse files
committed
Fix missing return character for impute_missing_values function
1 parent 556bcec commit e5938a5

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/sasctl/pzmm/write_score_code.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -805,7 +805,7 @@ def _predict_method(
805805
cls.score_code += f"\n{'':4})\n"
806806
if missing_values:
807807
cls.score_code += (
808-
f"{'':4}input_array = impute_missing_values(input_array)"
808+
f"{'':4}input_array = impute_missing_values(input_array)\n"
809809
)
810810
cls.score_code += (
811811
f"{'':4}column_types = {column_types}\n"
@@ -847,7 +847,7 @@ def _predict_method(
847847
cls.score_code += f"\n{'':4})\n"
848848
if missing_values:
849849
cls.score_code += (
850-
f"{'':4}input_array = impute_missing_values(input_array)"
850+
f"{'':4}input_array = impute_missing_values(input_array)\n"
851851
)
852852
cls.score_code += (
853853
f"{'':4}prediction = model.{method.__name__}(input_array)\n"
@@ -868,7 +868,7 @@ def _predict_method(
868868
cls.score_code += f"\n{'':4})\n"
869869
if missing_values:
870870
cls.score_code += (
871-
f"{'':4}input_array = impute_missing_values(input_array)"
871+
f"{'':4}input_array = impute_missing_values(input_array)\n"
872872
)
873873
cls.score_code += (
874874
f"{'':4}prediction = model.{method.__name__}(input_array)\n\n"
@@ -900,7 +900,7 @@ def _predict_method(
900900
cls.score_code += f"\n{'':4})\n"
901901
if missing_values:
902902
cls.score_code += (
903-
f"{'':4}input_array = impute_missing_values(input_array)"
903+
f"{'':4}input_array = impute_missing_values(input_array)\n"
904904
)
905905
cls.score_code += (
906906
f"{'':4}prediction = model.{method.__name__}(input_array).tolist()\n"

0 commit comments

Comments
 (0)