style: format code with black

Author: Satvik-Singh192

src/quant_research_starter/cli.py

@@ -131,7 +131,7 @@ def compute_factors(data_file, factors, output):
     "--plotly",
     is_flag=True,
     default=False,
-    help="Also generate interactive Plotly HTML chart"
+    help="Also generate interactive Plotly HTML chart",
 )
 def backtest(data_file, signals_file, initial_capital, output, plot, plotly):
     """Run backtest with given signals."""
@@ -229,7 +229,7 @@ def backtest(data_file, signals_file, initial_capital, output, plot, plotly):
             portfolio_values=results_dict["portfolio_value"],
             initial_capital=initial_capital,
             output_path=str(html_path),
-            plot_type="html"
+            plot_type="html",
         )
 
         click.echo(f"Plotly HTML chart saved -> {html_path}")

src/quant_research_starter/metrics/__init__.py

@@ -3,4 +3,4 @@
 from .plotting import create_equity_curve_plot
 from .risk import RiskMetrics
 
-__all__ = [ "create_equity_curve_plot","RiskMetrics"]
+__all__ = ["create_equity_curve_plot", "RiskMetrics"]

src/quant_research_starter/metrics/plotting.py

@@ -7,6 +7,7 @@
 
 try:
     import plotly.graph_objects as go
+
     PLOTLY_AVAILABLE = True
 except ImportError:
     PLOTLY_AVAILABLE = False
@@ -19,7 +20,7 @@ def create_equity_curve_plot(
     portfolio_values: List[float],
     initial_capital: float,
     output_path: str,
-    plot_type: str = "png"
+    plot_type: str = "png",
 ) -> str:
     """
     Create equity curve plot in specified format.
@@ -40,50 +41,54 @@ def create_equity_curve_plot(
         return _create_plotly_chart(portfolio_series, initial_capital, output_path)
     else:
         return _create_matplotlib_chart(portfolio_series, output_path)
+
+
 def _create_plotly_chart(
-    portfolio_series: pd.Series,
-    initial_capital: float,
-    output_path: str
+    portfolio_series: pd.Series, initial_capital: float, output_path: str
 ) -> str:
     """Create interactive Plotly HTML chart."""
     fig = go.Figure()
-    fig.add_trace(go.Scatter(
-        x=portfolio_series.index,
-        y=portfolio_series.values,
-        mode='lines',
-        name='Portfolio Value',
-        line={'color': '#2E86AB', 'width': 3},
-        hovertemplate='<b>Date</b>: %{x|%Y-%m-%d}<br><b>Value</b>: $%{y:,.2f}<extra></extra>'
-    ))
+    fig.add_trace(
+        go.Scatter(
+            x=portfolio_series.index,
+            y=portfolio_series.values,
+            mode="lines",
+            name="Portfolio Value",
+            line={"color": "#2E86AB", "width": 3},
+            hovertemplate="<b>Date</b>: %{x|%Y-%m-%d}<br><b>Value</b>: $%{y:,.2f}<extra></extra>",
+        )
+    )
     # Add initial capital reference line
     fig.add_hline(
         y=initial_capital,
         line_dash="dash",
         line_color="red",
         annotation_text=f"Initial Capital: ${initial_capital:,.0f}",
-        annotation_position="bottom right"
+        annotation_position="bottom right",
     )
     total_return_pct = ((portfolio_series.iloc[-1] / initial_capital) - 1) * 10
     fig.update_layout(
         title=f"Backtest Performance (Total Return: {total_return_pct:+.1f}%)",
-        xaxis_title='Date',
-        yaxis_title='Portfolio Value ($)',
-        template='plotly_white',
-        hovermode='x unified',
-        height=500
+        xaxis_title="Date",
+        yaxis_title="Portfolio Value ($)",
+        template="plotly_white",
+        hovermode="x unified",
+        height=500,
     )
-    fig.update_yaxes(tickprefix='$', tickformat=',.0f')
+    fig.update_yaxes(tickprefix="$", tickformat=",.0f")
     fig.write_html(output_path)
     return output_path
+
+
 def _create_matplotlib_chart(portfolio_series: pd.Series, output_path: str) -> str:
     """Create static matplotlib PNG chart."""
     plt.figure(figsize=(10, 6))
     plt.plot(portfolio_series.index, portfolio_series.values, linewidth=2)
-    plt.title('Portfolio Value')
-    plt.ylabel('USD')
+    plt.title("Portfolio Value")
+    plt.ylabel("USD")
     plt.grid(True, alpha=0.3)
     plt.gcf().autofmt_xdate()
     plt.tight_layout()
-    plt.savefig(output_path, dpi=150, bbox_inches='tight')
+    plt.savefig(output_path, dpi=150, bbox_inches="tight")
     plt.close()
     return output_path

tests/test_plotting.py

@@ -15,12 +15,12 @@ def test_plotly_html_creation():
         portfolio_values=portfolio_values,
         initial_capital=1000000,
         output_path=test_html_path,
-        plot_type="html"
+        plot_type="html",
     )
 
     # Verify file was created
     assert os.path.exists(html_path)
-    assert html_path.endswith('.html')
+    assert html_path.endswith(".html")
     assert os.path.getsize(html_path) > 1000
 
     # Cleanup
@@ -42,11 +42,11 @@ def test_plotly_fallback_to_matplotlib():
         portfolio_values=portfolio_values,
         initial_capital=1000000,
         output_path=test_png_path,
-        plot_type="png"
+        plot_type="png",
     )
 
     assert os.path.exists(png_path)
-    assert png_path.endswith('.png')
+    assert png_path.endswith(".png")
 
     # Cleanup
     if os.path.exists(png_path):