Include MultiState import in PySBFlatExporter (#453)

pysb · Jul 12, 2019 · f107091 · f107091
1 parent 4a4aee5
commit f107091
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pysb/export/pysb_flat.py b/pysb/export/pysb_flat.py
@@ -63,7 +63,7 @@ def write_cset(cset):
         output.write("\n")
         output.write("from pysb import Model, Monomer, Parameter, Expression, "
                      "Compartment, Rule, Observable, Initial, MatchOnce, "
-                     "Annotation, ANY, WILD\n")
+                     "Annotation, MultiState, ANY, WILD\n")
         output.write("\n")
         output.write("Model()\n")
         output.write("\n")