10 lines
254 B
Python
10 lines
254 B
Python
import pytest
|
|
|
|
from xgboost.testing.federated import run_federated_learning
|
|
|
|
|
|
@pytest.mark.parametrize("with_ssl", [True, False])
|
|
@pytest.mark.mgpu
|
|
def test_federated_learning(with_ssl: bool) -> None:
|
|
run_federated_learning(with_ssl, True, __file__)
|