config¶

class pyrosetta.distributed.cluster.config.EnvironmentConfig¶

Bases: Generic[G]

_ENV_VAR: str = 'PYROSETTACLUSTER_ENVIRONMENT_MANAGER'¶

_ENV_MANAGERS: Tuple[str, ...] = ('pixi', 'uv', 'mamba', 'conda')¶

_ENV_EXPORT_CMDS: Dict[str, str] = {'conda': "conda env export --prefix '/home/benchmark/working_dir/.venv'", 'mamba': "mamba env export --prefix '/home/benchmark/working_dir/.venv'", 'pixi': 'pixi lock --check || pixi lock --no-install', 'uv': 'uv export --format requirements-txt --frozen'}¶

__init__() → None¶

property env_export_cmd: str¶: Return the appropriate environment export command for the given environment manager. This method automatically adjusts for pixi and uv when a manifest path or project path is set via environment variables.

_is_protocol = False¶

pyrosetta.distributed.cluster.config.get_environment_config() → EnvironmentConfig¶: Return an instance of the EnvironmentConfig class on the host process.

pyrosetta.distributed.cluster.config.get_environment_manager() → str¶: Get the configured environment manager.

pyrosetta.distributed.cluster.config.get_environment_cmd() → str¶: Get the configured environment export command.

pyrosetta.distributed.cluster.config.get_environment_var() → str¶: Get the PyRosettaCluster operating system environment variable name.