Prevent from changing the interface specification once set.

Summary: Prevent from changing the interface specification once set. Differential Revision: D23502188 fbshipit-source-id: b21a8e53bb265a001e7a2772afee64a10e3d2c6e
facebookresearch · Sep 3, 2020 · c9b04a5 · c9b04a5
1 parent e413df4
commit c9b04a5
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: