change details

Author: dunzeng

docs/source/contributing.rst

@@ -56,6 +56,7 @@ Pull Request Checklist
             │   └── server                  # server related implementations
             │       └── scale               # scale manager
             │           └── hierarchical    # hierarchical communication pattern modules
+            ├── compressor                  # compressor modules
             └── utils                       # functional modules
                 └── dataset                 # functional modules associated with dataset
 

fedlab/utils/dataset/functional.py

@@ -77,7 +77,7 @@ def lognormal_unbalance_partition(num_clients, num_samples, unbalance_sgm):
 
 
 def hetero_dir_partition(targets, num_clients, num_classes, dir_alpha, min_require_size=None):
-    r"""
+    """
 
     Non-iid partition based on Dirichlet distribution. The method is from "hetero-dir" partition of
     `Bayesian Nonparametric Federated Learning of Neural Networks <https://arxiv.org/abs/1905.12022>`_