Add check to SCV import

hmorimitsu · Sep 13, 2021 · f55e59d · f55e59d
1 parent 788a8c9
commit f55e59d
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/ptlflow/__init__.py b/ptlflow/__init__.py
@@ -49,11 +49,17 @@
 from ptlflow.models.pwcnet.pwcnet import PWCNet, PWCDCNet
 from ptlflow.models.raft.raft import RAFT, RAFTSmall
 from ptlflow.models.scopeflow.irr_pwc_v2 import ScopeFlow
-from ptlflow.models.scv.scv import SCVEighth, SCVQuarter
 from ptlflow.models.starflow.starflow import StarFlow
 from ptlflow.models.vcn.vcn import VCN, VCNSmall
 from ptlflow.utils.utils import config_logging
 
+try:
+    from ptlflow.models.scv.scv import SCVEighth, SCVQuarter
+except ImportError as e:
+    print(e)
+    SCVEighth = None
+    SCVQuarter = None
+
 config_logging()