generate_input() additions

Reed-CompBio · Sep 11, 2024 · 8934b4c · 8934b4c
1 parent fc7d746
commit 8934b4c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/spras/omicsintegrator1.py b/spras/omicsintegrator1.py
@@ -49,7 +49,7 @@ def write_conf(filename=Path('config.txt'), w=None, b=None, d=None, mu=None, noi
 class OmicsIntegrator1(PRM):
     required_inputs = ['prizes', 'edges', 'dummy_nodes']
 
-    #TODO: edit generate_inputs to include NODEID (check notes)
+    #TODO: TEST !!!!! + test files??
 
     @staticmethod
     def generate_inputs(data, filename_map):
@@ -94,7 +94,7 @@ def generate_inputs(data, filename_map):
         if node_df.contains_node_columns('dummy'):
             dummy_df = node_df[node_df['dummy'] == True] 
             # save as list of dummy nodes
-            dummy_df.to_csv(filename_map['dummy_nodes'], sep='\t', index=False, columns=['NODEID'], header=None)
+            dummy_df.to_csv(filename_map['dummy_nodes'], sep='\t', index=False, columns=['NODEID', 'dummy'], header=['NODEID', 'dummy'])
         else:
             # create empty dummy file 
             with open(filename_map['dummy_nodes'], mode='w') as file: