diff --git a/datasets/flwr_datasets/federated_dataset.py b/datasets/flwr_datasets/federated_dataset.py index bdee9b5a3b37..9470e8b657a4 100644 --- a/datasets/flwr_datasets/federated_dataset.py +++ b/datasets/flwr_datasets/federated_dataset.py @@ -20,8 +20,8 @@ import datasets from datasets import Dataset, DatasetDict from flwr_datasets.partitioner import Partitioner -from flwr_datasets.telemetry import EventType, event from flwr_datasets.resplitter import Resplitter +from flwr_datasets.telemetry import EventType, event from flwr_datasets.utils import ( _check_if_dataset_tested, _instantiate_partitioners, diff --git a/datasets/flwr_datasets/telemetry.py b/datasets/flwr_datasets/telemetry.py index 45cb44817bfa..7de3dc4b8d69 100644 --- a/datasets/flwr_datasets/telemetry.py +++ b/datasets/flwr_datasets/telemetry.py @@ -114,9 +114,7 @@ class EventType(str, Enum): # The type signature is not compatible with mypy, pylint and flake8 # so each of those needs to be disabled for this line. # pylint: disable-next=no-self-argument,arguments-differ,line-too-long - def _generate_next_value_( - name: str, start: int, count: int, last_values: List[Any] - ) -> Any: # type: ignore # noqa: E501 + def _generate_next_value_(name: str, start: int, count: int, last_values: List[Any]) -> Any: # type: ignore # noqa: E501 return name PING = auto()