debug

app.py

@@ -142,6 +142,7 @@ def mask_out(input_ids,pron_locs,option_locs,mask_id):
     # note annotations are shifted by 1 because special tokens were omitted
     return input_ids[:pron_locs[0]+1] + [mask_id for _ in range(len(option_locs))] + input_ids[pron_locs[-1]+2:]
 
+'''
 def run_intervention(interventions,batch_size,model,masked_ids_option_1,masked_ids_option_2,option_1_tokens,option_2_tokens,pron_locs):
     probs = []
     for masked_ids, option_tokens in zip([masked_ids_option_1, masked_ids_option_2],[option_1_tokens,option_2_tokens]):
@@ -158,6 +159,7 @@ def run_intervention(interventions,batch_size,model,masked_ids_option_1,masked_i
     probs = np.array(probs)
     assert probs.shape[0]==2 and probs.shape[1]==2 and probs.shape[2]==batch_size
     return probs
+'''
 
 if __name__=='__main__':
     wide_setup()