Update the import paths

adap · May 28, 2024 · f52dc53 · f52dc53
1 parent b7facf4
commit f52dc53
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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):