Try to skip missing type checking on inputs

mypy.ini

@@ -0,0 +1,2 @@
+[teaal.*]
+ignore_missing_imports = True

tests/integration/demo.yaml

@@ -1,9 +1,26 @@
 einsum:
   declaration: # Ranks are listed alphabetically in this section
     TS: [T, P1, P0, E]
-    ND: [P1, P0, E]
+    ND: [P1, P0]
+    NTS: [T, P1, P0, E]
+    PF: [P1, P0, E, C]
+    OF: [P1, E]
+    PS: [P1, P0, E, C]
+    TOP: [P1, E, C]
   expressions:
-    - ND[p1, p0, e] = TS[t,  p1, p0, e]
+    - ND[p1, p0] = TS[t,  p1, p0, e]
+    - NTS[t,  p1, p0, e] = TS[t,  p1, p0, e] * ND[p1, p0]
+    - PF[p1, p0, e, c] = NTS[t,  p1, p0, e]
+    - OF[p1, e] = PF[p1, p0, e, c]
+    - PS[p1, p0, e, c] = NTS[t,  p1, p0, e] * OF[p1, e]
+    - TOP[p1, e, c] = PF[p1, p0, e, c] + PS[p1, p0, e, c]
 mapping:
+  rank-order:
+    PF: [P1, E, P0, C]
+    PS: [P1, E, P0, C]
   loop-order:
     ND: [T, P1, P0, E]
+    PF: [T, P1, E, P0, C]
+    OF: [P1, E, P0, C]
+    PS: [T, P1, E, P0, C]
+    TOP: [P1, E, P0, C]