trying fixing tests

owkin · Aug 12, 2024 · 244b894 · 244b894
1 parent 31fa318
commit 244b894
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/fedeca/utils/substrafl_utils.py b/fedeca/utils/substrafl_utils.py
@@ -366,10 +366,11 @@ def reset_experiment(self):
         self.test_data_nodes = None
         # We need to avoid persistence of DB in between runs, this is an obscure
         # hack but it's working
-        database = self.ds_client._backend._db._db._data
-        if len(database.keys()) > 1:
-            for k in list(database.keys()):
-                database.pop(k)
+        if hasattr(self.ds_client, "_backend"):
+            database = self.ds_client._backend._db._db._data
+            if len(database.keys()) > 1:
+                for k in list(database.keys()):
+                    database.pop(k)
 
 
 def get_outmodel_function(