From 9371c4f0e738763238ede8e99788f19770b8cb92 Mon Sep 17 00:00:00 2001 From: Vighnesh Birodkar Date: Wed, 6 Aug 2014 22:38:18 +0530 Subject: [PATCH] Fixed potential import error --- skimage/graph/_ncut.py | 6 +++++- skimage/graph/graph_cut.py | 2 +- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/skimage/graph/_ncut.py b/skimage/graph/_ncut.py index fbefca27..46ffb528 100644 --- a/skimage/graph/_ncut.py +++ b/skimage/graph/_ncut.py @@ -1,4 +1,8 @@ -import networkx as nx +try: + import networkx as nx +except ImportError: + import warnings + warnings.warn('RAGs require networkx') import numpy as np from scipy import sparse from . import _ncut_cy diff --git a/skimage/graph/graph_cut.py b/skimage/graph/graph_cut.py index 65d14585..e44f8a5c 100644 --- a/skimage/graph/graph_cut.py +++ b/skimage/graph/graph_cut.py @@ -2,7 +2,7 @@ try: import networkx as nx except ImportError: import warnings - warnings.warn('"cut_threshold" requires networkx') + warnings.warn('RAGs require networkx') import numpy as np from . import _ncut from . import _ncut_cy