removed unused import

ml-explore · Feb 9, 2024 · e548f8f · e548f8f
1 parent 57eb900
commit e548f8f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/python/mlx/nn/layers/upsample.py b/python/mlx/nn/layers/upsample.py
@@ -1,6 +1,6 @@
 # Copyright © 2023-2024 Apple Inc.
 
-from typing import List, Literal, Tuple, Union
+from typing import Literal, Tuple, Union
 
 import mlx.core as mx
 from mlx.nn.layers.base import Module
@@ -103,7 +103,9 @@ def __init__(
     ):
         super().__init__()
         if mode not in ["nearest", "bilinear"]:
-            raise ValueError(f"[Upsample2d] Got unsupported upsampling algorithm: {mode}")
+            raise ValueError(
+                f"[Upsample2d] Got unsupported upsampling algorithm: {mode}"
+            )
         if isinstance(scale, (list, tuple)):
             self.scale = tuple(map(float, scale))
         else: