Update code/process_input.py

code/process_input.py

@@ -20,6 +20,9 @@ def clean_data(input_set):
 
         elif '.txt' in input_set:
             data = pd.read_csv(input_set, sep='\t', names=['uniprotID', 'wt', 'pos', 'mut'])
+
+        else:
+            data =  pd.DataFrame(columns = ['uniprotID', 'wt', 'pos', 'mut'])
         data = data[['uniprotID', 'wt', 'pos', 'mut']]
 
         # Exclude termination codons, synonymous mutations and any non-standard residues such as Sec, 4 or 6.