From f52dc53ec6b40dd24e1e2c1c089bf84976f4d353 Mon Sep 17 00:00:00 2001 From: Adam Narozniak Date: Tue, 28 May 2024 09:51:50 +0200 Subject: [PATCH] Update the import paths --- datasets/flwr_datasets/federated_dataset.py | 2 +- datasets/flwr_datasets/partitioner/partitioner.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/datasets/flwr_datasets/federated_dataset.py b/datasets/flwr_datasets/federated_dataset.py index 3ab74006a94a..4db5de3e8255 100644 --- a/datasets/flwr_datasets/federated_dataset.py +++ b/datasets/flwr_datasets/federated_dataset.py @@ -21,7 +21,7 @@ from datasets import Dataset, DatasetDict from flwr_datasets.partitioner import Partitioner from flwr_datasets.resplitter import Resplitter -from flwr_datasets.telemetry import EventType, event +from flwr_datasets.common.telemetry import EventType, event from flwr_datasets.utils import ( _check_if_dataset_tested, _instantiate_partitioners, diff --git a/datasets/flwr_datasets/partitioner/partitioner.py b/datasets/flwr_datasets/partitioner/partitioner.py index 37fa7102c186..f5491303fbe2 100644 --- a/datasets/flwr_datasets/partitioner/partitioner.py +++ b/datasets/flwr_datasets/partitioner/partitioner.py @@ -19,7 +19,7 @@ from typing import Optional from datasets import Dataset -from flwr_datasets.telemetry import EventType, event +from flwr_datasets.common.telemetry import EventType, event class Partitioner(ABC):