Pull out immutable arg (#28)

narumiruna · Dec 6, 2019 · fd8d2dd · fd8d2dd
1 parent cc171f4
commit fd8d2dd
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/mlconfig/collections.py b/mlconfig/collections.py
@@ -29,16 +29,16 @@ def __setattr__(self, key, value):
         else:
             self[key] = value
 
-    def set_immutable(self):
-        self.__dict__[AttrDict.IMMUTABLE] = True
+    def set_immutable(self, immutable=True):
+        self.__dict__[AttrDict.IMMUTABLE] = immutable
 
         for value in self.__dict__.values():
             if isinstance(value, AttrDict):
-                value.set_immutable()
+                value.set_immutable(immutable)
 
         for value in self.values():
             if isinstance(value, AttrDict):
-                value.set_immutable()
+                value.set_immutable(immutable)
 
     def is_immutable(self):
         return self.__dict__[AttrDict.IMMUTABLE]