From: Jérôme Benoit <jerome.benoit@piment-noir.org>
Date: Fri, 14 Mar 2025 14:24:44 +0000 (+0100)
Subject: refactor(qav3): use iloc everywhere
X-Git-Url: https://git.piment-noir.org/?a=commitdiff_plain;h=bcddd5be9747a6cdf9a33ddc9e82e6fd961578a6;p=freqai-strategies.git

refactor(qav3): use iloc everywhere

Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>
---

diff --git a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py
index 1cf89f2..641a23e 100644
--- a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py
+++ b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py
@@ -174,7 +174,9 @@ class LightGBMRegressorQuickAdapterV35(BaseRegressionModel):
                 warmed_up = False
 
         pred_df_full = (
-            self.dd.historic_predictions[pair].tail(num_candles).reset_index(drop=True)
+            self.dd.historic_predictions[pair]
+            .iloc[-num_candles:]
+            .reset_index(drop=True)
         )
 
         if not warmed_up:
diff --git a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py
index 0e65884..f616f52 100644
--- a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py
+++ b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py
@@ -177,7 +177,9 @@ class XGBoostRegressorQuickAdapterV35(BaseRegressionModel):
                 warmed_up = False
 
         pred_df_full = (
-            self.dd.historic_predictions[pair].tail(num_candles).reset_index(drop=True)
+            self.dd.historic_predictions[pair]
+            .iloc[-num_candles:]
+            .reset_index(drop=True)
         )
 
         if not warmed_up: