0.3.2 bug_fixes for automatic download in create_dataset

senseppi/commands/create_dataset.py

@@ -230,8 +230,6 @@ class STRINGDatasetCreation:
         os.remove(self.intermediate_file)
         os.remove("clusters_preprocessed.tsv")
         os.remove("clusters.tsv")
-        os.remove(self.interactions_file)
-        os.remove(self.sequences_file)
 
     # A method to remove sequences of inappropriate length from a fasta file
     def process_fasta_file(self):
@@ -282,7 +280,9 @@ def add_args(parser):
 
 
 def main(params):
+    downloaded_flag = False
     if params.interactions is None or params.sequences is None:
+        downloaded_flag = True
         logging.info('One or both of the files are not specified (interactions or sequences). '
                      'Downloading from STRING...')
 
@@ -320,6 +320,10 @@ def main(params):
     data.final_preprocessing_positives()
     data.create_negatives()
 
+    if downloaded_flag:
+        os.remove(params.interactions)
+        os.remove(params.sequences)
+
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()