guohanghui commited on
Commit
a004336
·
verified ·
1 Parent(s): c8852c0

Update pyfolio/mcp_output/mcp_plugin/mcp_service.py

Browse files
pyfolio/mcp_output/mcp_plugin/mcp_service.py CHANGED
@@ -79,9 +79,16 @@ def _save_plot_as_base64(plt, plot_name):
79
  def _convert_to_series(data: List[float], name: str = "returns") -> pd.Series:
80
  """Convert list to pandas Series with date index."""
81
  if isinstance(data, list):
 
 
 
 
 
 
 
82
  return pd.Series(
83
  data,
84
- index=pd.date_range("2020-01-01", periods=len(data), freq="D"),
85
  name=name
86
  )
87
  return data
 
79
  def _convert_to_series(data: List[float], name: str = "returns") -> pd.Series:
80
  """Convert list to pandas Series with date index."""
81
  if isinstance(data, list):
82
+ # Create a business day date range to avoid NaT issues
83
+ try:
84
+ date_index = pd.date_range("2020-01-01", periods=len(data), freq="B")
85
+ except Exception:
86
+ # Fallback to daily frequency if business day fails
87
+ date_index = pd.date_range("2020-01-01", periods=len(data), freq="D")
88
+
89
  return pd.Series(
90
  data,
91
+ index=date_index,
92
  name=name
93
  )
94
  return data