Remove checks because they will be added to module.py

karandwivedi42 · karandwivedi42 · commit 302e5ed7219a · 2018-06-28T12:21:40.000-04:00
diff --git a/torch/nn/modules/container.py b/torch/nn/modules/container.py
@@ -226,9 +226,6 @@ def __getitem__(self, key):
         return self._modules[key]
 
     def __setitem__(self, key, module):
-        if not isinstance(key, torch._six.string_classes):
-            raise TypeError("ModuleDict.__setitem__ key should be string, but got " +
-                            type(key).__name__)
         self.add_module(key, module)
 
     def __delitem__(self, key):
@@ -432,9 +429,6 @@ def __getitem__(self, key):
         return self._parameters[key]
 
     def __setitem__(self, key, parameter):
-        if not isinstance(key, torch._six.string_classes):
-            raise TypeError("ParameterDict.__setitem__ key should be string,"
-                            " but got " + type(key).__name__)
         self.register_parameter(key, parameter)
 
     def __delitem__(self, key):
