Add an exists method (#328)

facebookresearch · Jun 9, 2022 · d19075c · d19075c
1 parent e24aac5
commit d19075c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/nle/dataset/db.py b/nle/dataset/db.py
@@ -21,6 +21,10 @@ def db(conn=None, filename=DB, new=False, rw=None, **kwargs):
             conn.close()
 
 
+def exists(filename=DB):
+    return os.path.exists(filename)
+
+
 def connect(filename=DB, new=False, rw=None, **kwargs):
     assert new ^ os.path.exists(filename)
     if not new and not rw: