From 48f28d4b0cb5628ff31f32529a85907cd81a1627 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Fri, 13 Dec 2019 12:42:54 -0500
Subject: [PATCH] fix
---
.../make_multiallelic_training_data.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
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 998d8dd4..4360a215 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(
--
GitLab