use default factories for dataclass attributes

dottxt-ai · Sep 17, 2024 · 2464eca · 2464eca
1 parent a3fe2b8
commit 2464eca
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/prompts/tokens.py b/prompts/tokens.py
@@ -1,4 +1,4 @@
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 from typing import Dict, Optional
 
 
@@ -10,10 +10,10 @@ class Limits:
 
 @dataclass
 class Special:
-    sequence: Limits = Limits("", "")
-    user: Limits = Limits("", "")
-    assistant: Limits = Limits("", "")
-    system: Limits = Limits("", "")
+    sequence: Limits = field(default_factory=lambda: Limits())
+    user: Limits = field(default_factory=lambda: Limits())
+    assistant: Limits = field(default_factory=lambda: Limits())
+    system: Limits = field(default_factory=lambda: Limits())
 
 
 SPECIAL_TOKENS: Dict[Optional[str], Special] = {