Merge branch 'feature/ecg_rpeak_vgraph_detector' of https://github.co…

…m/sokolmarek/NeuroKit into feature/ecg_rpeak_vgraph_detector

neurokit2/ecg/ecg_findpeaks.py

@@ -1068,7 +1068,7 @@ def _ecg_findpeaks_vgraph(signal, sampling_rate=1000, lowcut=3, order=2, **kwarg
     """
     # Try loading ts2vg
     try:
-        from ts2vg import NaturalVG
+        import ts2vg
     except ImportError as import_error:
         raise ImportError(
             "NeuroKit error: ecg_findpeaks(): the 'ts2vg' module is required for"
@@ -1098,7 +1098,7 @@ def _ecg_findpeaks_vgraph(signal, sampling_rate=1000, lowcut=3, order=2, **kwarg
         y = filtered[L:R]
 
         # Compute the adjacency matrix to the directed visibility graph
-        A = NaturalVG(directed="top_to_bottom").build(y).adjacency_matrix()
+        A = ts2vg.NaturalVG(directed="top_to_bottom").build(y).adjacency_matrix()
         _w = np.ones(len(y))
 
         # Computee the weights for the ecg using its VG transformation