expose public apis

Xiao-Chenguang · Oct 23, 2024 · 24f665c · 24f665c
1 parent e47ef49
commit 24f665c
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/fedmind/__init__.py b/fedmind/__init__.py
@@ -0,0 +1,11 @@
+from .algs.fedavg import FedAvg
+from .server import FedAlg
+from .utils import EasyDict, StateDict
+
+
+__all__ = [
+    "FedAvg",
+    "FedAlg",
+    "EasyDict",
+    "StateDict",
+]