Fixed circular dependency.

AgentSmith7 · Apr 13, 2021 · 3575edf · 3575edf
1 parent c44c052
commit 3575edf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/data.py b/data.py
@@ -27,7 +27,7 @@
 import os
 import requests
 from data_keys import *
-from returns import total_return
+import returns
 
 ########################################################################
 
@@ -344,7 +344,7 @@ def load_index_data(ticker, sales=True, book_value=True, dividend_TTM=True):
     # Calculate the Total Return.
     # The price-data from Yahoo does not contain the Total Return
     # for stock indices because it does not reinvest dividends.
-    df[TOTAL_RETURN] = total_return(df=df)
+    df[TOTAL_RETURN] = returns.total_return(df=df)
 
     # Load Sales Per Share data.
     if sales: