From d5c5e6e786db2a8e47cc1f5b3cefae9d7db24278 Mon Sep 17 00:00:00 2001 From: Ben Cornett Date: Fri, 12 Oct 2018 17:55:47 -0400 Subject: [PATCH] Simplify set logic --- indel_prediction/predictor/predict.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/indel_prediction/predictor/predict.py b/indel_prediction/predictor/predict.py index 8c34932..2f3605d 100644 --- a/indel_prediction/predictor/predict.py +++ b/indel_prediction/predictor/predict.py @@ -61,10 +61,10 @@ def predictMutations(theta_file, target_seq, pam_idx, add_null=True): feature_data, feature_columns = readFeaturesData(tmp_features_file) os.remove(tmp_features_file) - if len(set(theta_feature_columns).difference(set(feature_columns))) != 0: + if not set(feature_columns) >= set(theta_feature_columns): raise Exception('Stored feature names associated with model thetas are not contained in those computed') - if len(set(theta_feature_columns).union(set(feature_columns))) != len(theta_feature_columns): + if not set(theta_feature_columns) >= set(feature_columns): feature_data = feature_data[['Indel'] + theta_feature_columns] feature_columns = theta_feature_columns