fix typing for python<3.11

bmtgoncalves · bmtgoncalves · commit 5cf3887060be · 2025-03-30T11:19:04.000-04:00
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -23,6 +23,7 @@ dependencies = [
  "pyyaml (>=6.0.2,<7.0.0)",
  "seaborn (>=0.13.2,<0.14.0)",
  "scikit-learn (>=1.6.1,<2.0.0)",
+ "typing-extensions (>=4.13.0,<5.0.0)",
 ]
 [project.urls]
 Homepage = "https://github.com/DataForScience/epidemik"
diff --git a/src/epidemik/EpiModel.py b/src/epidemik/EpiModel.py
@@ -3,7 +3,13 @@
 # @author Bruno Goncalves
 ######################################################
 
-from typing import Dict, List, Set, Union, Self
+from typing import Dict, List, Set, Union
+
+try:
+    from typing import Self
+except ImportError:
+    from typing_extensions import Self
+
 import warnings
 import string
 import time

Original file line number	Diff line number	Diff line change
`@@ -23,6 +23,7 @@ dependencies = [`
`23`	`23`	`"pyyaml (>=6.0.2,<7.0.0)",`
`24`	`24`	`"seaborn (>=0.13.2,<0.14.0)",`
`25`	`25`	`"scikit-learn (>=1.6.1,<2.0.0)",`
	`26`	`+ "typing-extensions (>=4.13.0,<5.0.0)",`
`26`	`27`	`]`
`27`	`28`	`[project.urls]`
`28`	`29`	`Homepage = "https://github.com/DataForScience/epidemik"`