AlienChen commited on
Commit
7397fee
·
verified ·
1 Parent(s): a620d8f

Update utils/parsing.py

Browse files
Files changed (1) hide show
  1. utils/parsing.py +3 -0
utils/parsing.py CHANGED
@@ -25,6 +25,9 @@ def parse_guidance_args():
25
  parser.add_argument("--weights", type=float, nargs='+', required=False)
26
  parser.add_argument("--output_file", type=str, default='moo_outputs.txt')
27
  parser.add_argument("--motif_penalty", action='store_true')
 
 
 
28
 
29
  args = parser.parse_args()
30
  return args
 
25
  parser.add_argument("--weights", type=float, nargs='+', required=False)
26
  parser.add_argument("--output_file", type=str, default='moo_outputs.txt')
27
  parser.add_argument("--motif_penalty", action='store_true')
28
+ parser.add_argument("--objectives", nargs="+", type=str, required=False, default=None,
29
+ choices=["Hemolysis","Non-Fouling","Solubility","Half-Life","Affinity","Motif","Specificity"],
30
+ )
31
 
32
  args = parser.parse_args()
33
  return args