Include option to allow reloading modules without stage name clashes

LSSTDESC · Apr 8, 2022 · 641c611 · 641c611
1 parent e52e956
commit 641c611
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ceci/stage.py b/ceci/stage.py
@@ -38,6 +38,7 @@ class PipelineStage:
     dask_parallel = False
     config_options = {}
     doc = ""
+    allow_reload = False
 
     def __init__(self, args, comm=None):
         """Construct a pipeline stage, specifying the inputs, outputs, and configuration for it.
@@ -252,7 +253,7 @@ def __init_subclass__(cls, **kwargs):
 
         if stage_is_complete:
             # Deal with duplicated class names
-            if cls.name in cls.pipeline_stages:
+            if cls.name in cls.pipeline_stages and not cls.allow_reload:
                 other = cls.pipeline_stages[cls.name][1]
                 raise errors.DuplicateStageName(
                     "You created two pipeline stages with the"