fix(mosaic): drop Literal typing

loiccoyle · Jun 24, 2024 · 9fe6c0f · 9fe6c0f
1 parent 7eca6ba
commit 9fe6c0f
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/phomo/mosaic.py b/phomo/mosaic.py
@@ -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: