Merge branch 'master' into dask-env

PrefectHQ · Aug 9, 2019 · 03508ae · 03508ae
2 parents 5ca17af + 383c475
commit 03508ae
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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)
 

diff --git 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):