Merge pull request #13 from coqui-ai/copy-support

Copy support

Author: erogol

VERSION

@@ -1 +1 @@
-0.0.9
+0.0.10

coqpit/coqpit.py

@@ -5,7 +5,7 @@
 import os
 from collections.abc import MutableMapping
 from dataclasses import MISSING as _MISSING
-from dataclasses import Field, asdict, dataclass, fields, is_dataclass
+from dataclasses import Field, asdict, dataclass, fields, is_dataclass, replace
 from pathlib import Path
 from pprint import pprint
 from typing import Any, Dict, Generic, List, Optional, Type, TypeVar, Union, get_type_hints
@@ -558,20 +558,8 @@ def _is_initialized(self):
         at the initialization when no attribute has been defined."""
         return "_initialized" in vars(self) and self._initialized
 
-    def __setattr__(self, name: str, value: Any) -> None:
-        if self._is_initialized() and issubclass(type(value), Coqpit):
-            self.__fields__[name].type = type(value)
-        return super().__setattr__(name, value)
-
-    def __set_fields(self):
-        """Create a list of fields defined at the object initialization"""
-        self.__fields__ = {}  # pylint: disable=attribute-defined-outside-init
-        for field in fields(self):
-            self.__fields__[field.name] = field
-
     def __post_init__(self):
         self._initialized = True
-        self.__set_fields()
         try:
             self.check_values()
         except AttributeError:
@@ -583,7 +571,7 @@ def __iter__(self):
         return iter(asdict(self))
 
     def __len__(self):
-        return len(self.__fields__)
+        return len(fields(self))
 
     def __setitem__(self, arg: str, value: Any):
         setattr(self, arg, value)
@@ -645,6 +633,9 @@ def check_values(self):
     def has(self, arg: str) -> bool:
         return arg in vars(self)
 
+    def copy(self):
+        return replace(self)
+
     def update(self, new: dict, allow_new=False) -> None:
         """Update Coqpit fields by the input ```dict```.
 

tests/test_copying.py

@@ -0,0 +1,25 @@
+import copy
+from dataclasses import dataclass
+
+from coqpit.coqpit import Coqpit
+
+
+@dataclass
+class SimpleConfig(Coqpit):
+    val_a: int = 10
+
+
+def test_copying():
+    config = SimpleConfig()
+
+    config_new = config.copy()
+    config_new.val_a = 1234
+    assert config.val_a != config_new.val_a
+
+    config_new = copy.copy(config)
+    config_new.val_a = 4321
+    assert config.val_a != config_new.val_a
+
+    config_new = copy.deepcopy(config)
+    config_new.val_a = 4321
+    assert config.val_a != config_new.val_a

tests/test_simple_config.py

@@ -56,6 +56,7 @@ def test_simple_config():
     print(dict(**config))
 
     # value assignment by mapping
+    # TODO: MAYBE this should raise an errorby the value check.
     config["val_a"] = -999
     print(config["val_a"])
     assert config.val_a == -999