-
Notifications
You must be signed in to change notification settings - Fork 842
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add Driver
class
#2531
Add Driver
class
#2531
Conversation
src/py/flwr/driver/driver.py
Outdated
) | ||
return list(res.nodes) | ||
|
||
def push_task_ins(self, task_ins_list: Iterable[TaskIns]) -> List[str]: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Let's use List
here to be consistent with the argument name (we can generalize to Iterable
later):
def push_task_ins(self, task_ins_list: Iterable[TaskIns]) -> List[str]: | |
def push_task_ins(self, task_ins_list: List[TaskIns]) -> List[str]: |
src/py/flwr/driver/driver.py
Outdated
# Check if task_ids is None | ||
if task_ids is None: | ||
task_ids = list(self.task_id_pool) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Let's remove this utility for now, and leave it up to the user to keep track of task_ids
.
Add
Driver
class indriver/driver.py
.Add unit tests for
Driver
indriver/driver_test.py