Skip to content
Snippets Groups Projects
Commit ffae04d7 authored by Hans-Jörg's avatar Hans-Jörg
Browse files

Fix handling of CSV input in optimize script.

I think this did not work before.
parent 35129c52
No related branches found
No related tags found
No related merge requests found
......@@ -11,6 +11,7 @@ from schedgen.opt.schedule import (
from schedgen.opt.order import OptimizeOrder
from schedgen.opt.helper import message, file_or_dir_path, new_file_path, file_path
from schedgen.parsers.gridtpt import GridTPT
from schedgen.parsers.csvresult import CSVResult
args_parser = argparse.ArgumentParser(
description="Find an optimal schedule from a set of bechmarks."
......@@ -215,13 +216,19 @@ if __name__ == "__main__":
time_slices = args.slices
time_slices.sort()
message(f"Time slices: {time_slices}", logics)
experiments = list(args.data.glob("**/*.txt"))
message(f"Found {len(experiments)} .txt files.", logics)
if not args.csv:
experiments = list(args.data.glob("**/*.txt"))
message(f"Found {len(experiments)} .txt files.", logics)
if "all" in logics:
r = GridTPT(args.data)
filter = None
else:
r = GridTPT(args.data, logics)
filter = logics
if args.csv:
r = CSVResult(args.data, filter)
else:
r = GridTPT(args.data, filter)
exps = r.frame.copy()
pre_schedule = None
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment