diff --git a/test/functional/test_main.py b/test/functional/test_main.py index b5f179daa1a174849c4a1409f03010d11261fe94..986140da9b529578cd124daf1a2ce9f5394939b5 100644 --- a/test/functional/test_main.py +++ b/test/functional/test_main.py @@ -275,7 +275,7 @@ def run_test_case( ) -@pytest.mark.parametrize("strategy", ["FedAvg", "FedAvgM", "Scaffold"]) +@pytest.mark.parametrize("strategy", ["FedAvg", "Scaffold"]) @pytest.mark.parametrize("framework", FRAMEWORKS) @pytest.mark.parametrize("kind", ["Reg", "Bin", "Clf"]) @pytest.mark.filterwarnings("ignore: PyTorch JSON serialization") @@ -294,7 +294,7 @@ def test_declearn( Note: If websockets is unavailable, use gRPC (warn) or fail. """ if not fulltest: - if (kind != "Reg") or (strategy == "FedAvgM"): + if (kind != "Reg") or (strategy == "FedAvg"): pytest.skip("skip scenario (no --fulltest option)") protocol = "websockets" # type: Literal["grpc", "websockets"] if "websockets" not in list_available_protocols():