Make template copyable

LukasZahradnik · Nov 23, 2023 · a51d6fd · a51d6fd
1 parent 7d06df9
commit a51d6fd
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/neuralogic/core/template.py b/neuralogic/core/template.py
@@ -183,3 +183,14 @@ def __setitem__(self, key, value):
         if isinstance(value, (Iterable, Module)):
             raise NotImplementedError
         self.template[key] = value
+
+    def __copy__(self) -> "Template":
+        temp = Template()
+
+        temp.template_file = self.template_file
+        temp.template = self.template
+
+        return temp
+
+    def clone(self) -> "Template":
+        return self.__copy__()
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.black]
 line-length = 120
-target-version = ['py38']
+target-version = ['py39']
 
 [tool.mypy]
 ignore_missing_imports = true