ensure compatibility across python versions

DV-Anh · Nov 21, 2024 · 92cf92d · 92cf92d
1 parent daf3b50
commit 92cf92d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/custom/environments/customMPE.py b/custom/environments/customMPE.py
@@ -4,7 +4,8 @@
 import numpy as np
 import scipy.linalg as lin
 import chex
-from typing import Tuple, Dict, override
+from typing import Tuple, Dict
+from typing_extensions import override
 from flax import struct
 from functools import partial
 from jaxmarl.environments.mpe.simple import SimpleMPE, State, AGENT_COLOUR, OBS_COLOUR

diff --git a/requirements/requirements.txt b/requirements/requirements.txt
@@ -24,6 +24,7 @@ pillow>=10.2.0
 pettingzoo>=1.24.3
 tqdm>=4.66.0
 scipy<=1.12
+typing-extensions>=4.12
 
 requests>=2.32.3
 psycopg2-binary>=2.9.9