From 6782ec667ba622ea1dff1c21bfced04f427e8f81 Mon Sep 17 00:00:00 2001 From: Maxim Zemskov Date: Fri, 9 Aug 2019 17:18:54 +0300 Subject: [PATCH 1/2] Add quotes --- src/prefect/core/flow.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/prefect/core/flow.py b/src/prefect/core/flow.py index e713b07251c0..12be1a375222 100644 --- a/src/prefect/core/flow.py +++ b/src/prefect/core/flow.py @@ -1043,7 +1043,7 @@ def visualize( except ImportError: msg = ( "This feature requires graphviz.\n" - "Try re-installing prefect with `pip install prefect[viz]`" + "Try re-installing prefect with `pip install 'prefect[viz]'`" ) raise ImportError(msg) From 606e93ac9cb813d665bd97e6bdd69355446100d9 Mon Sep 17 00:00:00 2001 From: Maxim Zemskov Date: Fri, 9 Aug 2019 18:03:16 +0300 Subject: [PATCH 2/2] Update test_flow.py --- tests/core/test_flow.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/core/test_flow.py b/tests/core/test_flow.py index 55b139b1b94b..fa0d757bb6e3 100644 --- a/tests/core/test_flow.py +++ b/tests/core/test_flow.py @@ -967,7 +967,7 @@ def test_visualize_raises_informative_importerror_without_graphviz( with monkeypatch.context() as m: m.setattr(sys, "path", "") - with pytest.raises(ImportError, match="pip install prefect\[viz\]"): + with pytest.raises(ImportError, match="pip install 'prefect\[viz\]'"): f.visualize() def test_viz_returns_graph_object_if_in_ipython(self):