From 24f665ca6d705ef2a3b7c30da16a69ce1a6b767f Mon Sep 17 00:00:00 2001
From: Xiao-Chenguang <973990275@qq.com>
Date: Wed, 23 Oct 2024 15:27:07 +0100
Subject: [PATCH] expose public apis

---
 fedmind/__init__.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/fedmind/__init__.py b/fedmind/__init__.py
index e69de29..b86a5f4 100644
--- 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",
+]