Merge branch 'aki-rm-cuda-sync' of https://github.com/pyg-team/pytorc…

…h-frame into aki-rm-cuda-sync
pyg-team · Aug 10, 2024 · a324a6c · a324a6c
2 parents 95869bc + e187f03
commit a324a6c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/torch_frame/nn/encoder/stype_encoder.py b/torch_frame/nn/encoder/stype_encoder.py
@@ -716,7 +716,9 @@ def __init__(
     def init_modules(self) -> None:
         super().init_modules()
         num_cols = len(self.stats_list)
-        self.emb_dim_list = [stats[StatType.EMB_DIM] for stats in self.stats_list]
+        self.emb_dim_list = [
+            stats[StatType.EMB_DIM] for stats in self.stats_list
+        ]
         self.weight_list = ParameterList([
             Parameter(torch.empty(emb_dim, self.out_channels))
             for emb_dim in self.emb_dim_list