AdditiveConfiguratorChain: Drop None entries

This allows `AdditiveConfiguratorChain(foo if bar else None)`.
moi90 · May 30, 2024 · c3e5aa3 · c3e5aa3
1 parent c39fd0b
commit c3e5aa3
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/experitur/core/configurators.py b/experitur/core/configurators.py
@@ -463,8 +463,12 @@ class AdditiveConfiguratorChain(BaseConfigurator):
         shuffle (bool, optional): Shuffle child configurations.
     """
 
-    def __init__(self, *configurators: BaseConfigurator, shuffle=False) -> None:
-        self.configurators = configurators
+    def __init__(
+        self, *configurators: Optional[BaseConfigurator], shuffle=False
+    ) -> None:
+        # Drop None entries
+        # (This allows `AdditiveConfiguratorChain(foo if bar else None)`.)
+        self.configurators = [c for c in configurators if c is not None]
         self.shuffle = shuffle
 
     def build_sampler(