Fix black issues

2021-05-17 10:31:26 +08:00
parent 06fe246d82
commit b11cfe263d
8 changed files with 106 additions and 28 deletions
--- a/exps/LFNA/lfna_meta_model.py
+++ b/exps/LFNA/lfna_meta_model.py
@@ -75,7 +75,8 @@ class LFNA_Meta(super_core.SuperModule):

        # unknown token
        self.register_parameter(
-            "_unknown_token", torch.nn.Parameter(torch.Tensor(1, time_embedding)),
+            "_unknown_token",
+            torch.nn.Parameter(torch.Tensor(1, time_embedding)),
        )

        # initialization