From 079fb3d1710c1b8ea19740b80969acb9a4dab0d7 Mon Sep 17 00:00:00 2001 From: Stephen Oman Date: Tue, 29 Oct 2024 08:00:35 +0000 Subject: [PATCH] Fix to dataset delete syntax (#37) --- nle/dataset/db.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nle/dataset/db.py b/nle/dataset/db.py index 9d30347f3..1571f287d 100644 --- a/nle/dataset/db.py +++ b/nle/dataset/db.py @@ -162,8 +162,8 @@ def create_dataset(dataset_name, root, ttyrec_version=0, conn=None, commit=True) def delete_dataset(dataset_name, conn=None, commit=True): with db(conn, rw=True) as conn: - conn.execute("DELETE datasets WHERE dataset_name=?", (dataset_name,)) - conn.execute("DELETE roots WHERE dataset_name=?", (dataset_name,)) + conn.execute("DELETE FROM datasets WHERE dataset_name=?", (dataset_name,)) + conn.execute("DELETE FROM roots WHERE dataset_name=?", (dataset_name,)) conn.execute("UPDATE meta SET mtime = ?", (time.time(),)) if commit: conn.commit()