fix(mosaic): drop Literal typing

loiccoyle · Jun 24, 2024 · 41feff5 · 41feff5
1 parent 7eca6ba
commit 41feff5
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/phomo/mosaic.py b/phomo/mosaic.py
@@ -2,7 +2,7 @@
 import math
 from functools import partial
 from multiprocessing.pool import Pool as MpPool
-from typing import Literal, Tuple, Union
+from typing import Tuple, Union
 
 import numpy as np
 from PIL import Image
@@ -145,9 +145,7 @@ def d_matrix(
         self._log.debug("d_matrix shape: %s", d_matrix.shape)
         return d_matrix
 
-    def d_matrix_cuda(
-        self, metric: Literal["norm"] | Literal["greyscale"] = "norm"
-    ) -> np.ndarray:
+    def d_matrix_cuda(self, metric: str = "norm") -> np.ndarray:
         """Compute the distance matrix using CUDA for GPU acceleration.
 
         Args: