Skip to content

Commit

Permalink
Add __main__.py
Browse files Browse the repository at this point in the history
  • Loading branch information
danjjl committed Nov 14, 2024
1 parent c41640b commit a522cf8
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 1 deletion.
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
[project]
name = "szcore-evaluation"
version = "0.0.4"
version = "0.0.5"
description = "Compare szCORE compliant annotations of EEG datasets of people with epilelpsy."
authors = [
{ name = "Jonathan Dan", email = "[email protected]" }
Expand Down
13 changes: 13 additions & 0 deletions src/szcore_evaluation/__main__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
if __name__ == "__main__":
import argparse
from szcore_evaluation.evaluate import evaluate_dataset

parser = argparse.ArgumentParser(
description="Compare szCORE compliant annotations of EEG datasets of people with epilelpsy."
)
parser.add_argument("ref", help="Path to the folder containing the reference TSV files.")
parser.add_argument("hyp", help="Path to the folder containing the hypothesis TSV files.")
parser.add_argument("output", help="Path to the output JSON file where the results are saved.")

args = parser.parse_args()
evaluate_dataset(args.ref, args.hyp, args.output)

0 comments on commit a522cf8

Please sign in to comment.