final update

ersilia-os · Mar 23, 2024 · a1b9480 · a1b9480
1 parent 2ebc593
commit a1b9480
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/model/framework/code/main.py b/model/framework/code/main.py
@@ -46,6 +46,7 @@ def my_model(smiles_list):
 # write output in a .csv file
 with open(output_file, "w") as f:
     writer = csv.writer(f)
-    writer.writerow((["activity_10", "activity_20", "activity_40", "activity_60", "activity_80", "activity_100"]))  # header
+    writer.writerow((["activity"]))  # header
     for o in outputs:
-        writer.writerow(o)
+        writer.writerow([o[0]])
+