Skip to content

Commit c8be42a

Browse files
committed
Properly fix column_types for h2o score code (too many curly brackets).
1 parent 2e1fa34 commit c8be42a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/sasctl/pzmm/write_score_code.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -631,7 +631,7 @@ def _predict_method(
631631
f"{'':4}input_array = pd.DataFrame("
632632
f"[[{', '.join(var_list)}]],\n{'':31}columns=["
633633
f"{column_names}],\n{'':31}dtype=float,\n{'':31}"
634-
f"index=[0])\n{'':4}column_types = {{column_types}}\n"
634+
f"index=[0])\n{'':4}column_types = {column_types}\n"
635635
f"{'':4}h2o_array = h2o.H2OFrame(input_array, "
636636
f"column_types=column_types)\n{'':4}prediction = "
637637
f"model.{method.__name__}(h2o_array)\n{'':4}prediction"

0 commit comments

Comments
 (0)