fix: fixed suggestions

Co-authored-by: Copilot <[email protected]>

Author: Satvik-Singh192

src/quant_research_starter/cli.py

@@ -190,15 +190,21 @@ def run_with_progress(self, weight_scheme="rank"):
             )
             returns.append(ret)
 
-        results = pd.DataFrame({"returns": returns}, index=idx[1:])
-        results["portfolio_value"] = (
-            self.initial_capital * (1 + results["returns"]).cumprod()
+        results_df = pd.DataFrame({"returns": returns}, index=idx[1:])
+        results_df["portfolio_value"] = (
+            self.initial_capital * (1 + results_df["returns"]).cumprod()
         )
-        results["final_value"] = results["portfolio_value"].iloc[-1]
-        results["total_return"] = results["final_value"] / self.initial_capital - 1
-
-        return results
-
+        final_value = results_df["portfolio_value"].iloc[-1]
+        total_return = final_value / self.initial_capital - 1
+
+        # Return a dictionary matching the original structure
+        return {
+            "returns": results_df["returns"],
+            "portfolio_value": results_df["portfolio_value"],
+            "final_value": final_value,
+            "total_return": total_return,
+            # Add other keys as needed to match the original structure
+        }
     VectorizedBacktest.run = run_with_progress
 
     backtest = VectorizedBacktest(