Fix/imports (#21)

* atualiza pd.datetime (obsoleto) para datetime

* inclui yfinance e atualiza versão

setup.py

@@ -5,9 +5,9 @@
 
 setup(
     name="turingquant",
-    version="0.1.3",
+    version="0.1.4",
     packages=find_packages(),
-    install_requires=["pandas", "pandas_datareader", "numpy", "matplotlib", "alpha_vantage", "bs4", "plotly"],
+    install_requires=["pandas", "pandas_datareader", "numpy", "matplotlib", "alpha_vantage", "bs4", "plotly", "yfinance"],
 
     author="Grupo Turing",
     author_email="[email protected]",

turingquant/benchmark.py

@@ -1,19 +1,20 @@
 import pandas as pd
 import matplotlib.pyplot as plt
+from datetime import datetime
 from matplotlib.ticker import PercentFormatter
 from pandas_datareader import data
 
 
-def benchmark(ticker, start: pd.datetime, end: pd.datetime, source='yahoo', plot=True):
+def benchmark(ticker, start: datetime, end: datetime, source='yahoo', plot=True):
     """
     Essa função fornece um plot de retorno acumulado de um ativo ao longo de um dado intervalo de tempo, definido pelos parâmetros start e end.
     Os dados são coletados da API do yahoo, caso haja dados faltantes, os retornos são contabilizados como nulos.
 
     ticker[str]: recebe o ticker do papel que será obtido
 
-    start[pd.datetime]: início do intervalo
+    start[datetime]: início do intervalo
 
-    end[pd.datetime]: final do interval
+    end[datetime]: final do interval
 
     plot[bool]: (default = True) se True, realiza o plot
     """
@@ -26,27 +27,27 @@ def benchmark(ticker, start: pd.datetime, end: pd.datetime, source='yahoo', plot
     return cumulative
 
 
-def benchmark_ibov(start: pd.datetime, end: pd.datetime, source='yahoo', plot=True):
+def benchmark_ibov(start: datetime, end: datetime, source='yahoo', plot=True):
     """
     Essa função produz um plot da evolução do Índice Bovespa ao longo de um dado intervalo, definido pelos parâmetros start e end.
 
-    start[pd.datetime]: início do intervalo
+    start[datetime]: início do intervalo
 
-    end[pd.datetime]: final do interval
+    end[datetime]: final do interval
 
     plot[bool]: (default = True) se True, realiza o plot
     """
 
     return benchmark('^BVSP', start=start, end=end, source=source, plot=plot)
 
 
-def benchmark_sp500(start: pd.datetime, end: pd.datetime, source='yahoo', plot=True):
+def benchmark_sp500(start: datetime, end: datetime, source='yahoo', plot=True):
     """
     Essa função produz um plot da evolução do Índice S&P500 ao longo de um dado intervalo, definido pelos parâmetros start e end.
 
-    start[pd.datetime]: início do intervalo
+    start[datetime]: início do intervalo
 
-    end[pd.datetime]: final do interval
+    end[datetime]: final do interval
 
     plot[bool]: (default = True) se True, realiza o plot
     """