Catching absence of MAF info in VEP response

lib/postgap/Cisreg.py

@@ -306,12 +306,6 @@ def run(self, snps, tissues):
 		res = []
 		for hit in transcript_consequences:
 			for consequence in hit['transcript_consequences']:
-				MAFs = None
-				for variant in hit['colocated_variants']:
-					if variant['id'] == hit['id']:
-						 MAFs = variant
-						 break
-
 				res.append(Cisregulatory_Evidence(
 					snp = snp_hash[hit['input']],
 					gene = postgap.Ensembl_lookup.get_ensembl_gene(consequence['gene_id']),
@@ -321,7 +315,6 @@ def run(self, snps, tissues):
 					tissue = None,
 					info = {
 						'consequence_terms': consequence['consequence_terms'], 
-						'MAFs': MAFs
 					}
 				))
 

lib/postgap/Reg.py

@@ -178,23 +178,20 @@ def run(self, snps, tissues):
 		snp_hash = dict( (snp.rsID, snp) for snp in snps)
 		res = []
 		for hit in list:
-			MAFs = None
-			for variant in hit['colocated_variants']:
-				if variant['id'] == hit['id']:
-					 MAFs = variant
-					 break
-
-			if MAFs is not None:
-				res.append(Regulatory_Evidence(
-					snp = snp_hash[hit['input']],
-					score = 0,
-					source = self.display_name,
-					study = None,
-					tissue = None,
-					info = {
-						'MAFs': MAFs
-					}
-				))
+			if 'colocated_variants' in hit:
+				for variant in hit['colocated_variants']:
+					if variant['id'] == hit['id']:
+						res.append(Regulatory_Evidence(
+							snp = snp_hash[hit['input']],
+							score = 0,
+							source = self.display_name,
+							study = None,
+							tissue = None,
+							info = {
+								'MAFs': variant
+							}
+						))
+						break
 
 		self.logger.info("\tFound %i interactions in VEP" % (len(res)))