fix(mtnetwork): add import check

try to import HyPhy before use in transmission_network.fit_degree_distribution closes veg#40
FynnFreyer · Nov 8, 2023 · 2327969 · 2327969
1 parent ccd648a
commit 2327969
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/hivclustering/mtnetwork.py b/hivclustering/mtnetwork.py
@@ -2509,8 +2509,8 @@ def fit_degree_distribution(self, degree_option=None, hy_instance=None):
             try:
                 import hppy as hy
             except ImportError:
-                raise ImportError("Optional dependencies not found. please install the 'edgefiltering' extras to use this function: pip install hivclustering[edgefiltering]")
-
+                raise ImportError(
+                    "Optional dependencies not found. Please install the 'edgefiltering' extras to use this function: pip install hivclustering[edgefiltering]")
             hy_instance = hy.HyphyInterface()
 
         script_path = os.path.realpath(__file__)