Merge pull request #499 from IainHammond/master

Updated import in PCA skysubtraction
vortex-exoplanet · Mar 15, 2022 · 50e10eb · 50e10eb
2 parents f417541 + ee96f6a
commit 50e10eb
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/vip_hci/preproc/skysubtraction.py b/vip_hci/preproc/skysubtraction.py
@@ -44,7 +44,10 @@ def cube_subtract_sky_pca(sci_cube, sky_cube, mask, ref_cube=None, ncomp=2,
          - reconstructed cube.
 
     """
-    from ..pca.svd import svd_wrapper
+    try:
+        from ..psfsub.svd import svd_wrapper
+    except:
+        from ..pca.svd import svd_wrapper
 
     if sci_cube.shape[1] != sky_cube.shape[1] or sci_cube.shape[2] != \
             sky_cube.shape[2]: