Merge pull request #108 from CUNY-CL/varnames

Styles variable names
CUNY-CL · Jul 8, 2023 · 28ff0e5 · 28ff0e5
2 parents ed277fc + 5dcc2bb
commit 28ff0e5
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/yoyodyne/models/pointer_generator.py b/yoyodyne/models/pointer_generator.py
@@ -371,11 +371,11 @@ def forward(
 
     @staticmethod
     def _reshape_hiddens(
-        H: torch.Tensor, C: torch.Tensor, layers: int, num_directions: int
+        h: torch.Tensor, c: torch.Tensor, layers: int, num_directions: int
     ) -> Tuple[torch.Tensor, torch.Tensor]:
-        H = H.view(layers, num_directions, H.size(1), H.size(2)).sum(axis=1)
-        C = C.view(layers, num_directions, C.size(1), C.size(2)).sum(axis=1)
-        return (H, C)
+        h = h.view(layers, num_directions, h.size(1), h.size(2)).sum(axis=1)
+        c = c.view(layers, num_directions, c.size(1), c.size(2)).sum(axis=1)
+        return h, c
 
     @property
     def name(self) -> str: