Camel.toolkits.bohrium toolkit

​

BohriumToolkit

class BohriumToolkit(BaseToolkit):

• timeout (Optional[float], optional): The timeout for BohriumToolkit. (default: :obj:None)
• api_key (Optional[str], optional): The API key for Bohrium client. (default: :obj:None)
• project_id (Optional[int], optional): The project ID for Bohrium client. (default: :obj:None)
• yaml_path (Optional[str], optional): The path to the YAML file containing the job parameters. (default: :obj:None)

​

init

def __init__(
    self,
    timeout: Optional[float] = None,
    api_key: Optional[str] = None,
    project_id: Optional[int] = None,
    yaml_path: Optional[str] = None,
    _test_mode: bool = False
):

​

_custom_insert

def _custom_insert(self, data):

​

submit_job

def submit_job(
    self,
    job_name: str = 'bohr-job',
    machine_type: str = 'c2_m4_cpu',
    cmd: str = 'mpirun -n 2 lmp_mpi -i in.shear',
    image_address: str = 'registry.dp.tech/dptech/lammps:29Sep2021'
):

• job_name (str): The name of the job. It will be updated when yaml file is provided. The yaml file might be set when initialize BohriumToolkit. (default: :obj:bohr-job)
• machine_type (str): The type of machine to use. It will be updated when yaml file is provided. The yaml file might be set when initialize BohriumToolkit. (default: :obj:c2_m4_cpu)
• cmd (str): The command to run. It will be updated when yaml file is provided. The yaml file might be set when initialize (default: :obj:mpirun -n 2 lmp_mpi -i in.shear)
• image_address (str): The address of the image to use. It will be updated when yaml file is provided. The yaml file might be set when initialize BohriumToolkit. (default: :obj:registry.dp.tech/dptech/lammps:29Sep2021)

​

get_job_details

def get_job_details(self, job_id: int):

• job_id (int): The ID of the job.

​

terminate_job

def terminate_job(self, job_id: int):

• job_id (int): The ID of the job to terminate.

​

kill_job

def kill_job(self, job_id: int):

• job_id (int): The ID of the job to kill.

​

get_job_logs

def get_job_logs(
    self,
    job_id: int,
    log_file: str = 'STDOUTERR',
    page: int = -1,
    page_size: int = 8192
):

• job_id (int): The ID of the job.
• log_file (str, optional): The log file to get. (default: :obj:STDOUTERR)
• page (int, optional): The page number. (default: :obj:-1)
• page_size (int, optional): The page size. (default: :obj:8192)

​

create_job_group

def create_job_group(self, project_id: int, job_group_name: str):

• project_id (int): The ID of the project.
• job_group_name (str): The name of the job group.

​

download_job_results

def download_job_results(self, job_id: int, save_path: str):

• job_id (int): The ID of the job.
• save_path (str): The path to save the results to.

​

get_tools

def get_tools(self):