Regularize handling of dense features (#1447)

Summary: Pull Request resolved: #1447 Regularize handling of dense features Differential Revision: D23502163 fbshipit-source-id: 8657c38da27bfb2f6561ed0ba0b1f7417f29c36e
facebookresearch · Sep 3, 2020 · cc9abc6 · cc9abc6
1 parent 9ef7bf2
commit cc9abc6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pytext/torchscript/module.py b/pytext/torchscript/module.py
@@ -69,12 +69,12 @@ def __init__(
     @torch.jit.script_method
     def forward(
         self,
-        dense_feat: List[List[float]],
         texts: Optional[List[str]] = None,
         # multi_texts is of shape [batch_size, num_columns]
         multi_texts: Optional[List[List[str]]] = None,
         tokens: Optional[List[List[str]]] = None,
         languages: Optional[List[str]] = None,
+        dense_feat: Optional[List[List[float]]] = None,
     ):
         inputs: ScriptBatchInput = ScriptBatchInput(
             texts=resolve_texts(texts, multi_texts),
@@ -107,13 +107,13 @@ def __init__(
     @torch.jit.script_method
     def forward(
         self,
-        right_dense_feat: List[List[float]],
-        left_dense_feat: List[List[float]],
         texts: Optional[List[str]] = None,
         # multi_texts is of shape [batch_size, num_columns]
         multi_texts: Optional[List[List[str]]] = None,
         tokens: Optional[List[List[str]]] = None,
         languages: Optional[List[str]] = None,
+        right_dense_feat: Optional[List[List[float]]] = None,
+        left_dense_feat: Optional[List[List[float]]] = None,
     ):
         inputs: ScriptBatchInput = ScriptBatchInput(
             texts=resolve_texts(texts, multi_texts),