incorporated suggested changes

nu-radio · Dec 13, 2024 · 8557329 · 8557329
1 parent c149205
commit 8557329
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/NuRadioReco/modules/base/module.py b/NuRadioReco/modules/base/module.py
@@ -2,8 +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.RNO_G.rnog_detector
+import NuRadioReco.detector.detector as detectors
 import inspect
 import pickle
 
@@ -75,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: