Update app.py

app.py

@@ -788,8 +788,13 @@ if st.session_state.app_step == "dev":
         te = _prepare_table(book[sh_test].copy(),  "Testing",  st.session_state["FEATURES"], st.session_state["TARGET_BO"], st.session_state["TARGET_BD"], st.session_state["CANON_DEPTH"])
 
         # Predict with exact training feature order
-        tr[PRED_BO] = model_bo.predict(_make_X(tr, st.session_state["FEATURES"]))
-        tr[PRED_BD] = model_bd.predict(_make_X(tr, st.session_state["FEATURES"]))
+        Xtr = _make_X(tr, st.session_state["FEATURES"])  # only features
+        Xte = _make_X(te, st.session_state["FEATURES"])
+        tr[PRED_BO] = model_bo.predict(Xtr)
+        tr[PRED_BD] = model_bd.predict(Xtr)
+        te[PRED_BO] = model_bo.predict(Xte)
+        te[PRED_BD] = model_bd.predict(Xte)
+
         te[PRED_BO] = model_bo.predict(_make_X(te, st.session_state["FEATURES"]))
         te[PRED_BD] = model_bd.predict(_make_X(te, st.session_state["FEATURES"]))