NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-19 12:57:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[env]: add initial ProteinBinderEnv

Browse source 
Co-authored-by: based-tachikoma <based.tachikoma@gmail.com>
This commit is contained in:
hallerite 2025-05-18 14:38:29 -07:00
parent c189fc3351
commit 4d9bec44c6
No known key found for this signature in database
GPG key ID: 6D9E61BC2A1A58E1
4 changed files with 1245 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

33
environments/hack0/protein_design_env/configs/binderbench_default.yaml Normal file
View file

@ -0,0 +1,33 @@
# NVIDIA NIM Environment Default Configuration for BinderBench
# Debug Mode - set to true to use mock data instead of actual API calls
debug_protein_design_calls: false
# Retry settings for failed steps
max_retries_per_internal_step: 100 # Increased to allow many retries for tool calls
# API Settings
# nim_api_key is loaded from .env file using NVIDIA_NIM_API_KEY
nim_api_base_url: "https://health.api.nvidia.com/v1"
api_timeout: 600
polling_interval: 10
# Protein Design Settings
output_dir: "environments/hack0/protein_design_env/outputs"
# WandB tracking settings
use_wandb: true
wandb_name: "binderbench"
wandb_project: "atropos" # Will default to this if not specified
include_messages: true # Include messages in WandB logs
# Dataset configuration
dataset_name: "ronig/protein_binding_sequences"
target_col: "receptor"
binder_col: "peptide"
# Scoring weights for final complex quality
metric_weights:
plddt: 0.3
ptm: 0.3
iptm: 0.4