tier-4 SRT member: model.py

model.py

@@ -58,6 +58,11 @@ class ModMulBP(ModularMultiplicationModel):
             self.scratch = bool(ckpt.get("scratch", False))
             self.cursor = bool(ckpt.get("cursor", False))
             self.subpad = bool(ckpt.get("subpad", False))
+            self.stepidx = bool(ckpt.get("stepidx", False))
+            self.skiptriv = bool(ckpt.get("skiptriv", False))
+            self.subnum = bool(ckpt.get("subnum", False))
+            self.bemit = bool(ckpt.get("bemit", False))
+            self.srt = bool(ckpt.get("srt", False))
             self.regime = self.base ** self.W
         else:
             self.regime = 10 ** self.W
@@ -94,9 +99,12 @@ class ModMulBP(ModularMultiplicationModel):
         if self.composed:
             prompt_fn = lambda p, x, y, W: cenc.prompt_str(  # noqa: E731
                 x, y, p, W, self.base, self.subpad)
-            n_gen = cenc.gen_len(self.W, self.scratch, self.cursor, self.subpad)
+            n_gen = cenc.gen_len(self.W, self.scratch, self.cursor, self.subpad,
+                                 self.stepidx, self.skiptriv, self.subnum,
+                                 self.bemit, self.srt)
             decode_fn = lambda g, W: cenc.decode_answer(     # noqa: E731
-                g, W, self.base, self.scratch, self.cursor, self.subpad)
+                g, W, self.base, self.scratch, self.cursor, self.subpad,
+                self.stepidx, self.skiptriv, self.subnum, self.bemit, self.srt)
         else:
             cot = self.scratchpad or self.school
             prompt_fn = enc.prompt_str_sp if cot else enc.prompt_str