Merge pull request #780 from nu-radio/fix_import_statement_base_module

add missing import statement modules/base/module.py
nu-radio · Dec 13, 2024 · f3f146d · f3f146d
2 parents 279428a + 8557329
commit f3f146d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/NuRadioReco/modules/base/module.py b/NuRadioReco/modules/base/module.py
@@ -2,7 +2,7 @@
 from timeit import default_timer as timer
 import NuRadioReco.framework.event
 import NuRadioReco.framework.base_station
-import NuRadioReco.detector.detector_base
+import NuRadioReco.detector.detector as detectors
 import inspect
 import pickle
 
@@ -74,7 +74,7 @@ def register_run_method(self, *args, **kwargs):
                 # station should be second argument
                 elif isinstance(value, NuRadioReco.framework.base_station.BaseStation) and idx == 1:
                     station = value
-                elif isinstance(value, (NuRadioReco.detector.detector_base.DetectorBase, NuRadioReco.detector.RNO_G.rnog_detector.Detector)):
+                elif isinstance(value, (detectors.detector_base.DetectorBase, detectors.rnog_detector.Detector)):
                     pass # we don't try to store detectors
                 else: # we try to store other arguments IF they are pickleable
                     try: