update parse script

bigcode-project · Apr 14, 2024 · b480156 · b480156
1 parent 5b6eb9e
commit b480156
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/script/parse.py b/script/parse.py
@@ -281,7 +281,7 @@ def reconstruct_problem(data):
 
 def check_test_wo_doc(data):
     "Check if the problem is related to file system, network requests and database"
-    if "shutil" in data["libs"] or "requests" in data["libs"] or "django" in data["libs"] or "sqlite3" in data["libs"]:
+    if any([lib in data["libs"] for lib in ["shutil", "requests", "django", "sqlite3", "datetime"]]):
         return True
     elif "os.path" in data["apis"]:
         return True