From a9f14ac11944fa3e81115d80308339d372080bc2 Mon Sep 17 00:00:00 2001 From: Italo <45588475+italodamato@users.noreply.github.com> Date: Thu, 20 Jan 2022 18:44:09 +0000 Subject: [PATCH] show buy_tag only if not None --- freqtrade/plot/plotting.py | 6 +++--- tests/test_plotting.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/freqtrade/plot/plotting.py b/freqtrade/plot/plotting.py index b8a747105..8384c49b7 100644 --- a/freqtrade/plot/plotting.py +++ b/freqtrade/plot/plotting.py @@ -235,9 +235,9 @@ def plot_trades(fig, trades: pd.DataFrame) -> make_subplots: # Trades can be empty if trades is not None and len(trades) > 0: # Create description for sell summarizing the trade - trades['desc'] = trades.apply(lambda row: f"{row['profit_ratio']:.2%}, " - f"{row['buy_tag']}, " - f"{row['sell_reason']}, " + trades['desc'] = trades.apply(lambda row: f"{row['profit_ratio']:.2%}, " + \ + (f"{row['buy_tag']}, " if row['buy_tag'] is not None else "") + \ + f"{row['sell_reason']}, " + \ f"{row['trade_duration']} min", axis=1) trade_buys = go.Scatter( diff --git a/tests/test_plotting.py b/tests/test_plotting.py index 1ad292df0..6846bb5f2 100644 --- a/tests/test_plotting.py +++ b/tests/test_plotting.py @@ -187,7 +187,7 @@ def test_plot_trades(testdatadir, caplog): assert len(trades.loc[trades['profit_ratio'] <= 0]) == len(trade_sell_loss.x) assert trade_sell_loss.marker.color == 'red' assert trade_sell_loss.marker.symbol == 'square-open' - assert trade_sell_loss.text[5] == '-10.45%, None, stop_loss, 720 min' + assert trade_sell_loss.text[5] == '-10.45%, stop_loss, 720 min' def test_generate_candlestick_graph_no_signals_no_trades(default_conf, mocker, testdatadir, caplog):