Change | to Union

adap · Jul 15, 2024 · b7a8c12 · b7a8c12
1 parent 0618de9
commit b7a8c12
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/datasets/flwr_datasets/partitioner/distribution_partitioner.py b/datasets/flwr_datasets/partitioner/distribution_partitioner.py
@@ -16,7 +16,7 @@
 
 
 from collections import Counter
-from typing import Dict, List, Optional
+from typing import Dict, List, Optional, Union
 
 import numpy as np
 
@@ -120,7 +120,7 @@ class label distribution. `float` values are rounded to the nearest `int`.
 
     def __init__(  # pylint: disable=R0913
         self,
-        distribution_array: NDArrayInt | NDArrayFloat,
+        distribution_array: Union[NDArrayInt, NDArrayFloat],
         num_partitions: int,
         num_unique_labels_per_partition: int,
         partition_by: str,