Update compiler.py

timofey260 · Dec 11, 2021 · 432468f · 432468f
1 parent aff2931
commit 432468f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/compiler.py b/compiler.py
@@ -58,7 +58,10 @@ def cmp(code: list[str,]):  # code compiler
         line = m.split()
         try:  # if line be blank []
             if line[0] == "MARK":
-                marks[line[1]] = counter
+                if len(line) == 3:
+                    marks[line[1]] = int(line[2])
+                else:
+                    marks[line[1]] = counter
                 setzero()
             if line[0] in instructions:
                 counter += 1