From 1aa7d9627ed41d5952bc657d50c8774657aa2ff0 Mon Sep 17 00:00:00 2001 From: didier Date: Sat, 6 Mar 2021 18:20:27 +0000 Subject: [PATCH] Fix black failing test --- .../prediction_techniques/fbprophet.py | 1 + .../prediction_techniques/neural_networks.py | 12 ++---------- 2 files changed, 3 insertions(+), 10 deletions(-) diff --git a/gamestonk_terminal/prediction_techniques/fbprophet.py b/gamestonk_terminal/prediction_techniques/fbprophet.py index 3a93230555f..3f894feb7c9 100644 --- a/gamestonk_terminal/prediction_techniques/fbprophet.py +++ b/gamestonk_terminal/prediction_techniques/fbprophet.py @@ -14,6 +14,7 @@ register_matplotlib_converters() warnings.simplefilter("ignore") + def fbprophet(l_args, s_ticker, df_stock): parser = argparse.ArgumentParser( prog="fbprophet", diff --git a/gamestonk_terminal/prediction_techniques/neural_networks.py b/gamestonk_terminal/prediction_techniques/neural_networks.py index c062fd1ecf0..4fab5020788 100644 --- a/gamestonk_terminal/prediction_techniques/neural_networks.py +++ b/gamestonk_terminal/prediction_techniques/neural_networks.py @@ -8,21 +8,13 @@ from pandas.plotting import register_matplotlib_converters from TimeSeriesCrossValidation import splitTrain from sklearn.preprocessing import MinMaxScaler, StandardScaler from tensorflow.keras.models import Sequential -from tensorflow.keras.layers import ( - LSTM, - SimpleRNN, - Dense, - Dropout, - # Activation, - # RepeatVector, - # TimeDistributed, -) +from tensorflow.keras.layers import LSTM, SimpleRNN, Dense, Dropout from gamestonk_terminal.helper_funcs import ( check_positive, get_next_stock_market_days, parse_known_args_and_warn, - print_pretty_prediction + print_pretty_prediction, ) from gamestonk_terminal import config_neural_network_models as cfg_nn_models