diff --git a/src/dreamer/surprisal.py b/src/dreamer/surprisal.py index 093b2d7b..e05ce84d 100644 --- a/src/dreamer/surprisal.py +++ b/src/dreamer/surprisal.py @@ -261,7 +261,7 @@ async def _fetch_recent_observations( workspace_name=workspace_name, observer=observer, observed=observed, - filters={"level": levels} if levels else None, + filters={"level": {"in": levels}} if levels else None, limit=limit, )