From 2ba5b63e303681aea54fb831f9f681fe2a526aca Mon Sep 17 00:00:00 2001
From: Santosh Passoubady <61618641+ssantoshp@users.noreply.github.com>
Date: Wed, 7 Feb 2024 20:21:07 -0500
Subject: [PATCH] Update main.py

---
 src/empyrial/main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/empyrial/main.py b/src/empyrial/main.py
index 39c8cd4b..fbe0e79f 100644
--- a/src/empyrial/main.py
+++ b/src/empyrial/main.py
@@ -189,7 +189,7 @@ def graph_allocation(my_portfolio):
 
 
 def empyrial(my_portfolio, rf=0.0, sigma_value=1, confidence_value=0.95, report=False, filename="empyrial_report.pdf"):
-    if my_portfolio.rebalance != None:
+    if isinstance(my_portfolio.rebalance, pd.DataFrame):
         # we want to get the dataframe with the dates and weights
         rebalance_schedule = my_portfolio.rebalance