Skip to content

Commit 87eb32d

Browse files
Make sure tree specific filters handled correctly
1 parent 63f767f commit 87eb32d

1 file changed

Lines changed: 10 additions & 4 deletions

File tree

catalog/__init__.py

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -49,12 +49,18 @@ def games(**kwargs) -> pd.DataFrame:
4949
records: list[dict[str, Any]] = []
5050

5151
def check_filters(game: gbt.Game) -> bool:
52-
if "n_actions" in kwargs and len(game.actions) != kwargs["n_actions"]:
53-
return False
52+
if "n_actions" in kwargs:
53+
if not game.is_tree:
54+
return False
55+
if len(game.actions) != kwargs["n_actions"]:
56+
return False
5457
if "n_contingencies" in kwargs and len(game.contingencies) != kwargs["n_contingencies"]:
5558
return False
56-
if "n_infosets" in kwargs and len(game.infosets) != kwargs["n_infosets"]:
57-
return False
59+
if "n_infosets" in kwargs:
60+
if not game.is_tree:
61+
return False
62+
if len(game.infosets) != kwargs["n_infosets"]:
63+
return False
5864
if "is_const_sum" in kwargs and game.is_const_sum != kwargs["is_const_sum"]:
5965
return False
6066
if "is_perfect_recall" in kwargs and game.is_perfect_recall != kwargs["is_perfect_recall"]:

0 commit comments

Comments
 (0)