quick fix (#35)

owkin · Jun 22, 2024 · cf7a824 · cf7a824
1 parent 6c1497e
commit cf7a824
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/fedeca/fedeca_core.py b/fedeca/fedeca_core.py
@@ -263,8 +263,8 @@ def __init__(
             )
             "of clients for global bootstrap by passing num_clients"
             if bootstrap_function == "global":
-                assert self.total_size is not None, "You need to provide the total"
-                " size of the dataset for global bootstrap by setting total_size"
+                assert self.clients_sizes is not None, "You need to provide the total"
+                " size of the dataset for global bootstrap by setting clients_sizes"
                 print(
                     "WARNING: global bootstrap necessitates that the opener returns"
                     f" a column {self.client_identifier} with levels="