🔨 Fix TimeEmbed format

Author: marianovitasari20

ice_station_zebra/models/common/timeembed.py

@@ -5,8 +5,10 @@
 
 
 class TimeEmbed(nn.Module):
-    def __init__(self, dim: int = 256, 
-                 activation: str = "SiLU",) -> None:
+    def __init__(self, 
+                 dim: int = 256, 
+                 activation: str = "SiLU",
+    ) -> None:
         super().__init__()
 
         def act():