Skip to content

Commit

Permalink
passing drop_artifacts as argument (#81)
Browse files Browse the repository at this point in the history
  • Loading branch information
adriangutierrezg committed Nov 20, 2023
1 parent 53bcb91 commit 190251c
Showing 1 changed file with 10 additions and 1 deletion.
11 changes: 10 additions & 1 deletion tools/old_format_output.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,14 +91,23 @@ def convert_to_mat(groups, outfname, drop_artifacts=False):
def main(fname_data, fname_sorting, sign, outfname, drop_artifacts=False):
"""
read groups from sorting for conversion
Args:
drop_arftifacts, bool: flag for excluding artifacts. If True,
artifacts will be excluded. Defaults
to False.
"""
man = Combinato(fname_data, sign, fname_sorting)
# man.set_sign_times_spikes(sign)
# res = man.init_sorting(fname_sorting)
fn1 = os.path.basename(fname_data)
# fn2 = os.path.basename(fname_sorting)

joined = man.get_groups_joined()
if drop_artifacts:
joined = man.get_groups_joined(artifacts=False)
else:
joined = man.get_groups_joined(artifacts=True)

ch_name = man.header['AcqEntName']
# ugly: this works only if you use the CSCxy convention!
go_on = False
Expand Down

0 comments on commit 190251c

Please sign in to comment.