rename package to fedmind and use flat structure

src/flair/algs/fedavg.py → fedmind/algs/fedavg.py

@@ -2,8 +2,8 @@
 from torch.nn.modules.loss import _Loss
 from torch.utils.data import DataLoader
 
-from flair.server import FedAlg
-from flair.utils import EasyDict
+from fedmind.server import FedAlg
+from fedmind.utils import EasyDict
 
 
 class FedAvg(FedAlg):

src/flair/client.py → fedmind/client.py

@@ -11,7 +11,7 @@
 from torch.optim.optimizer import Optimizer
 from torch.utils.data import DataLoader
 
-from flair.utils import StateDict
+from fedmind.utils import StateDict
 
 
 def train(

src/flair/config.py → fedmind/config.py

@@ -2,7 +2,7 @@
 
 import yaml
 
-from flair.utils import EasyDict
+from fedmind.utils import EasyDict
 
 
 def dotodict(k: str, value):

src/flair/server.py → fedmind/server.py

@@ -10,8 +10,8 @@
 from torch.optim import SGD
 from torch.utils.data import DataLoader
 
-from flair.client import test, train, train_process
-from flair.utils import EasyDict, StateDict
+from fedmind.client import test, train, train_process
+from fedmind.utils import EasyDict, StateDict
 
 
 class FedAlg:
@@ -48,7 +48,7 @@ def __init__(
 
         self.wb_run = wandb.init(
             mode="offline",
-            project=args.get("WB_PROJECT", "flair"),
+            project=args.get("WB_PROJECT", "fedmind"),
             entity=args.get("WB_ENTITY", "wandb"),
             config=self.args.to_dict(),
             settings=wandb.Settings(_disable_stats=True, _disable_machine_info=True),

hello.py

@@ -1,5 +1,5 @@
 def main():
-    print("Hello from flair!")
+    print("Hello from fedmind!")
 
 
 if __name__ == "__main__":

pyproject.toml

@@ -1,5 +1,5 @@
 [project]
-name = "flair"
+name = "fedmind"
 version = "0.1.0"
 description = "Add your description here"
 readme = "README.md"
@@ -13,7 +13,7 @@ dependencies = [
 ]
 
 [project.scripts]
-new = "flair:main"
+new = "fedmind:main"
 
 [build-system]
 requires = ["hatchling"]

test/test_config.py

@@ -1,4 +1,4 @@
-from flair.config import get_config
+from fedmind.config import get_config
 import yaml
 
 

test/test_fedavg.py

@@ -6,9 +6,9 @@
 from torchvision.datasets import MNIST
 from torchvision.transforms import ToTensor
 
-from flair.algs.fedavg import FedAvg
-from flair.data import ClientDataset
-from flair.config import get_config
+from fedmind.algs.fedavg import FedAvg
+from fedmind.data import ClientDataset
+from fedmind.config import get_config
 
 
 def test_fedavg():