Commit d2a62a34 authored by vdberg's avatar vdberg
Browse files

arg to focus on is now a command line argument

parent cedeefa8
Loading
Loading
Loading
Loading
+2 −1
Original line number Diff line number Diff line
@@ -43,6 +43,7 @@ def load_nameslab():

parser = ArgumentParser()
parser.add_argument("-c", "--combine", action="store_true", default=False, help="whether to combine or not")
parser.add_argument("-a", "--arg", default='', help="focus on an arg")
args = parser.parse_args()
df = load_output(dir='outputs', combine=args.combine)

@@ -50,7 +51,7 @@ print('Output size {}:'.format(df.shape))

df = df[df['name'] != 'Ryan']

#df = df[df['arg'].str.startswith('B-ARG0')]
df = df[df['arg'].str.startswith(args.arg)]
#print('Arg-0 only {}:'.format(df.shape))