Make container slightly more flexible.

This change is fully backwards compatible.

Author: lucasb-eyer

DeepFried2/containers/Container.py

@@ -8,6 +8,7 @@ def __init__(self, *modules):
 
         self.modules = []
         for module in modules:
+            assert isinstance(module, Module), "`Container`s can only contain objects of (sub)type `Module`."
             self.add(module)
 
     def evaluate(self):
@@ -36,5 +37,5 @@ def get_stat_updates(self):
             stat_updates += module.get_stat_updates()
         return stat_updates
 
-    def add(self, module):
-        self.modules.append(module)
+    def add(self, *modules):
+        self.modules += modules