Update cleanup_lab.py

CederGroupHub · Mar 26, 2024 · 19a9951 · 19a9951
1 parent ed1b67b
commit 19a9951
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/alab_management/scripts/cleanup_lab.py b/alab_management/scripts/cleanup_lab.py
@@ -45,6 +45,10 @@ def cleanup_lab(
             print(f"Removing database {database_name}")
             _GetMongoCollection.init()
             _GetMongoCollection.client.drop_database(database_name)  # type: ignore
+        else:
+            print(f"Removing database {database_name}")
+            _GetMongoCollection.init()
+            _GetMongoCollection.client.drop_database(database_name)
 
         # if sim_mode != AlabOSConfig().is_sim_mode() or database_name == "Alab":
         #     print("Wrong name of database. Hence, not removed.")