gate-track-record-until-close

nifty_backend/__pycache__/runtime.cpython-311.pyc

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:15d55e363fc7f252b4827535702a967f81d0465b7649ef75739a960d176d2169
+size 117741

nifty_backend/runtime.py

@@ -1122,6 +1122,8 @@ def build_prediction_track_record(
         & daily_rows["close"].map(lambda value: np.isfinite(float(value)) if pd.notna(value) else False)
     ].copy()
     daily_rows = daily_rows[daily_rows["open"].astype(float) != 0]
+    completed_day = expected_completed_daily_date()
+    daily_rows = daily_rows[daily_rows["date"].dt.date <= completed_day]
     if daily_rows.empty:
         return []
 
@@ -1142,19 +1144,13 @@ def build_prediction_track_record(
             **(meta or {}),
         }
 
-    for _, row in t5_test.iterrows():
-        add_prediction(row.get("date"), row.get("prediction"), "T+5", 30, {"prob_up": row.get("prob_up")})
     for _, row in tomorrow_test.iterrows():
         pred = row.get("prediction")
         if pd.isna(pred) and "pred" in row:
             pred = "UP" if int(row.get("pred")) == 1 else "DOWN"
         add_prediction(row.get("target_date") or row.get("date"), pred, "Tomorrow", 20, {"prob_up": row.get("prob_up")})
-    for _, row in tplus1_test.iterrows():
-        add_prediction(row.get("target_date") or row.get("date"), row.get("prediction"), "T+1", 10, {"prob_up": row.get("prob_up")})
 
     add_prediction(tomorrow_latest.get("target_date"), tomorrow_latest.get("prediction"), "Tomorrow", 40, {"prob_up": tomorrow_latest.get("prob_up")})
-    add_prediction(tplus1_latest.get("target_date"), tplus1_latest.get("prediction"), "T+1", 35, {"prob_up": tplus1_latest.get("prob_up")})
-    add_prediction(t5_latest.get("input_date"), t5_latest.get("prediction"), "T+5", 45, {"prob_up": t5_latest.get("prob_up")})
 
     records: list[dict[str, Any]] = []
     for _, row in daily_rows.tail(20).iterrows():