Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 8 additions & 4 deletions nle/dataset/dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -211,29 +211,33 @@ def __init__(
SELECT ttyrecs.gameid, ttyrecs.part, ttyrecs.path
FROM ttyrecs
INNER JOIN datasets ON ttyrecs.gameid=datasets.gameid
WHERE datasets.dataset_name=?"""
WHERE datasets.dataset_name=?
ORDER BY ttyrecs.gameid"""

meta_sql = """
SELECT games.*
FROM games
INNER JOIN datasets ON games.gameid=datasets.gameid
WHERE datasets.dataset_name=?"""
WHERE datasets.dataset_name=?
ORDER BY games.gameid"""

if subselect_sql:
path_select = """
SELECT ttyrecs.gameid, ttyrecs.part, ttyrecs.path
FROM ttyrecs
INNER JOIN datasets ON ttyrecs.gameid=datasets.gameid
WHERE datasets.dataset_name=?
AND ttyrecs.gameid IN (%s)"""
AND ttyrecs.gameid IN (%s)
ORDER BY ttyrecs.gameid"""
core_sql = path_select % subselect_sql

meta_select = """
SELECT games.*
FROM games
INNER JOIN datasets ON games.gameid=datasets.gameid
WHERE datasets.dataset_name=?
AND games.gameid IN (%s)"""
AND games.gameid IN (%s)
ORDER BY games.gameid"""
meta_sql = meta_select % subselect_sql
sql_args = subselect_sql_args if subselect_sql_args else ()
sql_args = (dataset_name,) + sql_args
Expand Down
Loading