Make ROOT an optional import

DeepLearnPhysics · Sep 13, 2024 · ca784a3 · ca784a3
1 parent e3d295d
commit ca784a3
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/spine/io/read/larcv.py b/spine/io/read/larcv.py
@@ -1,9 +1,9 @@
 """Contains a reader class dedicated to loading data from LArCV files."""
 
 import numpy as np
-import ROOT
 
 from spine.utils.decorators import inherit_docstring
+from spine.utils.conditional import ROOT
 
 from .base import ReaderBase
 

diff --git a/spine/utils/conditional.py b/spine/utils/conditional.py
@@ -8,6 +8,12 @@
 import os
 from warnings import warn
 
+# If ROOT is available, load it
+try:
+    import ROOT
+except ModuleNotFoundError:
+    warn("ROOT could not be found, cannot parse LArCV data.")
+    ROOT = None
 
 # If LArCV is available, load it
 try: