✨v0.1.9 patch: ModelConfig

Supported assigning `extractor_config`, `head_config`, etc. explicitly.
carefree0910 · Dec 26, 2020 · f3df248 · f3df248
1 parent 08a8932
commit f3df248
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/cflearn/api/basic.py b/cflearn/api/basic.py
@@ -65,14 +65,26 @@ def __init__(
     def transform_config(self) -> Dict[str, Any]:
         return self.meta_config.setdefault("transform", {})
 
+    @transform_config.setter
+    def transform_config(self, value: Dict[str, Any]) -> None:
+        self.meta_config["transform"] = value
+
     @property
     def extractor_config(self) -> Dict[str, Any]:
         return self.meta_config.setdefault("extractor", {})
 
+    @extractor_config.setter
+    def extractor_config(self, value: Dict[str, Any]) -> None:
+        self.meta_config["extractor"] = value
+
     @property
     def head_config(self) -> Dict[str, Any]:
         return self.meta_config.setdefault("head", {})
 
+    @head_config.setter
+    def head_config(self, value: Dict[str, Any]) -> None:
+        self.meta_config["head"] = value
+
     def replace(self, **kwargs: Any) -> None:
         pipe_config_dict = self.pipe_config._asdict()
         pipe_config_dict.update(kwargs)