no

neggles · Jan 23, 2024 · 78d0408 · 78d0408
1 parent 445d69a
commit 78d0408
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/neurosis/models/autoencoder.py b/src/neurosis/models/autoencoder.py
@@ -35,11 +35,11 @@ class AbstractAutoencoder(L.LightningModule):
 
     def __init__(
         self,
-        ema_decay: Union[None, float] = None,
-        monitor: Union[None, str] = None,
+        ema_decay: Optional[float] = None,
+        monitor: Optional[str] = None,
         input_key: str = "jpg",
-        ckpt_path: Union[None, str] = None,
-        ignore_keys: Union[tuple, list] = tuple(),
+        ckpt_path: Optional[str] = None,
+        ignore_keys: tuple | list = tuple(),
         base_lr: Optional[float] = None,
     ):
         super().__init__()