diff --git a/xl2times/__main__.py b/xl2times/__main__.py index 0f7a0d5..1acdfcb 100644 --- a/xl2times/__main__.py +++ b/xl2times/__main__.py @@ -444,12 +444,15 @@ def convert_parameter(tablename: str, df: DataFrame): "ts.dd": ["ALL_TS"], "milestonyr.dd": ["MILESTONYR"], "output.dd": [ - t for t in config.dd_table_order if t not in ["ALL_TS", "MILESTONYR"] + t for t in config.dd_table_order if t not in {"ALL_TS", "MILESTONYR"} ], } for fname, tablenames in tables_in_file.items(): with open(os.path.join(output_dir, fname), "w", encoding=encoding) as fout: + # Include GAMS dollar control options + if fname == "output.dd": + fout.write("$ONEPS\n$ONWARNING\n\n") for tablename in [t for t in tablenames if t in tables]: df = tables[tablename] if tablename in sets: