diff --git a/downloads-generation/models_class1_pan_refined/make_multiallelic_training_data.py b/downloads-generation/models_class1_pan_refined/make_multiallelic_training_data.py index 998d8dd4a8eabfd39391a4c7e6bff6c3a5b7db8e..4360a215017c1fa6b7a190435529bedcb36bde7c 100644 --- a/downloads-generation/models_class1_pan_refined/make_multiallelic_training_data.py +++ b/downloads-generation/models_class1_pan_refined/make_multiallelic_training_data.py @@ -44,6 +44,7 @@ parser.add_argument( nargs="+", help="Include only the specified alleles") + def run(): args = parser.parse_args(sys.argv[1:]) hit_df = pandas.read_csv(args.hits) @@ -72,7 +73,7 @@ def run(): if args.alleles: filter_alleles = set(args.alleles) new_hit_df = hit_df.loc[ - hit_df.alleles.isin.map( + hit_df.alleles.map( lambda a: len(set(a).intersection(filter_alleles)) > 0) ] print(