Source code for p2pfl.learning.frameworks.simulation

"""Module for the efficient parallel local simulation of the learning process (based on Ray)."""

from p2pfl.learning.frameworks.learner import Learner
from p2pfl.utils.check_ray import ray_installed

###
#   Ray
###


[docs] def try_init_learner_with_ray(learner: Learner) -> Learner: """ Create a learner instance. Args: learner: The learner to wrap. """ if ray_installed(): from p2pfl.learning.frameworks.simulation.virtual_learner import VirtualNodeLearner learner = VirtualNodeLearner(learner) return learner