there is a Self after all

kima-org · Jun 27, 2024 · b2bf8c2 · b2bf8c2
1 parent 5aa6c19
commit b2bf8c2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -19,6 +19,7 @@ classifiers = [
 ]
 
 dependencies = [
+    "typing_extensions",
     "numpy",
     "scipy",
     "matplotlib",

diff --git a/src/kima/pykima/results.py b/src/kima/pykima/results.py
@@ -6,7 +6,8 @@
 import os
 import sys
 import pickle
-from typing import List, Union
+from typing import List, Union, Any
+from typing_extensions import Self
 import zipfile
 import time
 import tempfile
@@ -186,8 +187,8 @@ class posterior_holder:
     η5: np.ndarray = field(init=False)
     η6: np.ndarray = field(init=False)
     # 
-    tr = field(init=False)
-    ko = field(init=False)
+    tr: Self = field(init=False)
+    ko: Self = field(init=False)
 
     def __repr__(self):
         fields = list(self.__dataclass_fields__.keys())