Prevent from changing the interface specification once set.

Summary: Prevent from changing the interface specification once set. Differential Revision: D23502188 fbshipit-source-id: dbf5b28c220ec9845c010576e923e2369a2f4392
facebookresearch · Sep 3, 2020 · 9ef7bf2 · 9ef7bf2
1 parent a78effd
commit 9ef7bf2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pytext/torchscript/module.py b/pytext/torchscript/module.py
@@ -150,6 +150,8 @@ def inference_interface(self, argument_type: str):
         # LANGUAGES = 3
         # DENSE_FEAT = 4
 
+        if self.argno != -1:
+            raise RuntimeError("Cannot change argument type.")
         if argument_type == "texts":
             self.argno = TEXTS
         else: