[Minor] compatible with Python 3.8

ai4co · Mar 15, 2024 · 6a1adf2 · 6a1adf2
1 parent a8d806f
commit 6a1adf2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/rl4co/models/zoo/pomo/model.py b/rl4co/models/zoo/pomo/model.py
@@ -43,7 +43,7 @@ def __init__(
         baseline: str = "shared",
         num_augment: int = 8,
         augment_fn: Union[str, callable] = "dihedral8",
-        feats: list[str] = None,
+        feats: list = None,
         num_starts: int = None,
         select_start_nodes_fn: callable = select_start_nodes,
         **kwargs,

diff --git a/rl4co/models/zoo/symnco/model.py b/rl4co/models/zoo/symnco/model.py
@@ -42,7 +42,7 @@ def __init__(
         baseline: str = "symnco",
         num_augment: int = 4,
         augment_fn: Union[str, callable] = "symmetric",
-        feats: list[str] = None,
+        feats: list = None,
         alpha: float = 0.2,
         beta: float = 1,
         num_starts: int = 0,