aiida.orm.nodes.process.calculation package

Module with Node sub classes for calculation processes.

Submodules

Module with Node sub class for calculation function processes.

class aiida.orm.nodes.process.calculation.calcfunction.CalcFunctionNode(backend: StorageBackend | None = None, user: User | None = None, computer: Computer | None = None, **kwargs: Any)

基类：FunctionCalculationMixin, CalculationNode

ORM class for all nodes representing the execution of a calcfunction.

_CLS_NODE_LINKS

CalcFunctionNodeLinks 的别名

__abstractmethods__ = frozenset({})

__module__ = 'aiida.orm.nodes.process.calculation.calcfunction'

__parameters__ = ()

__plugin_type_string: ClassVar[str]

__query_type_string: ClassVar[str]

_abc_impl = <_abc._abc_data object>

_logger: Logger | None = <Logger aiida.orm.nodes.process.calculation.calcfunction.CalcFunctionNode (WARNING)>

fields: QbFields = {'attributes': 'QbDictField(attributes.*) -> Dict[str, Any]',  'computer_pk': 'QbNumericField(attributes.computer_pk) -> Optional[int]',  'ctime': 'QbNumericField(ctime) -> datetime',  'description': 'QbStrField(description) -> str',  'exception': 'QbStrField(attributes.exception) -> Optional[str]',  'exit_message': 'QbStrField(attributes.exit_message) -> Optional[str]',  'exit_status': 'QbNumericField(attributes.exit_status) -> Optional[int]',  'extras': 'QbDictField(extras.*) -> Dict[str, Any]',  'label': 'QbStrField(label) -> str',  'mtime': 'QbNumericField(mtime) -> datetime',  'node_type': 'QbStrField(node_type) -> str',  'paused': 'QbField(attributes.paused) -> bool',  'pk': 'QbNumericField(pk) -> int',  'process_label': 'QbStrField(attributes.process_label) -> Optional[str]',  'process_state': 'QbStrField(attributes.process_state) -> Optional[str]',  'process_status': 'QbStrField(attributes.process_status) -> Optional[str]',  'process_type': 'QbStrField(attributes.process_type) -> Optional[str]',  'repository_metadata': 'QbDictField(repository_metadata) -> Dict[str, Any]',  'sealed': 'QbField(attributes.sealed) -> bool',  'user_pk': 'QbNumericField(user_pk) -> int',  'uuid': 'QbStrField(uuid) -> str'}

class aiida.orm.nodes.process.calculation.calcfunction.CalcFunctionNodeLinks(node: Node)

基类：ProcessNodeLinks

Interface for links of a node instance.

__annotations__ = {}

__module__ = 'aiida.orm.nodes.process.calculation.calcfunction'

validate_outgoing(target: Node, link_type: LinkType, link_label: str) → None

Validate adding a link of the given type from ourself to a given node.

A calcfunction cannot return Data, so if we receive an outgoing link to a stored Data node, that means the user created a Data node within our function body and stored it themselves or they are returning an input node. The latter use case is reserved for @workfunctions, as they can have RETURN links.

参数:

• target – the node to which the link is going

• link_type – the link type

• link_label – the link label

抛出:

• TypeError – if target is not a Node instance or link_type is not a LinkType enum

• ValueError – if the proposed link is invalid

Module with Node sub class for calculation job processes.

class aiida.orm.nodes.process.calculation.calcjob.CalcJobNode(backend: StorageBackend | None = None, user: User | None = None, computer: Computer | None = None, **kwargs: Any)

基类：CalculationNode

ORM class for all nodes representing the execution of a CalcJob.

CALC_JOB_STATE_KEY = 'state'

IMMIGRATED_KEY = 'imported'

REMOTE_WORKDIR_KEY = 'remote_workdir'

RETRIEVE_LIST_KEY = 'retrieve_list'

RETRIEVE_TEMPORARY_LIST_KEY = 'retrieve_temporary_list'

SCHEDULER_DETAILED_JOB_INFO_KEY = 'detailed_job_info'

SCHEDULER_JOB_ID_KEY = 'job_id'

SCHEDULER_LAST_CHECK_TIME_KEY = 'scheduler_lastchecktime'

SCHEDULER_LAST_JOB_INFO_KEY = 'last_job_info'

SCHEDULER_STATE_KEY = 'scheduler_state'

_CLS_NODE_CACHING

CalcJobNodeCaching 的别名

__abstractmethods__ = frozenset({})

__module__ = 'aiida.orm.nodes.process.calculation.calcjob'

__parameters__ = ()

__plugin_type_string: ClassVar[str]

__qb_fields__: Sequence[QbField] = [QbStrField('scheduler_state', dtype=Optional[str], is_attribute=True), QbStrField('state', dtype=Optional[str], is_attribute=True), QbStrField('remote_workdir', dtype=Optional[str], is_attribute=True), QbStrField('job_id', dtype=Optional[str], is_attribute=True), QbStrField('scheduler_lastchecktime', dtype=Optional[str], is_attribute=True), QbStrField('last_job_info', dtype=Optional[str], is_attribute=True), QbDictField('detailed_job_info', dtype=Optional[dict], is_attribute=True), QbArrayField('retrieve_list', dtype=Optional[List[str]], is_attribute=True), QbArrayField('retrieve_temporary_list', dtype=Optional[List[str]], is_attribute=True), QbField('imported', dtype=Optional[bool], is_attribute=True)]

__query_type_string: ClassVar[str]

_abc_impl = <_abc._abc_data object>

_hash_ignored_attributes: Tuple[str, ...] = ('metadata_inputs', 'queue_name', 'account', 'qos', 'priority', 'max_wallclock_seconds', 'max_memory_kb')

_logger: Logger | None = <Logger aiida.orm.nodes.process.calculation.calcjob.CalcJobNode (WARNING)>

_tools = None

_updatable_attributes: Tuple[str, ...] = ('sealed', 'paused', 'checkpoints', 'exception', 'exit_message', 'exit_status', 'process_label', 'process_state', 'process_status', 'state', 'imported', 'remote_workdir', 'retrieve_list', 'retrieve_temporary_list', 'job_id', 'scheduler_state', 'scheduler_lastchecktime', 'last_job_info', 'detailed_job_info')

static _validate_retrieval_directive(directives: Sequence[str | Tuple[str, str, str]]) → None

Validate a list or tuple of file retrieval directives.

参数:

directives – a list or tuple of file retrieval directives

抛出:

ValueError – if the format of the directives is invalid

delete_state() → None

Delete the calculation job state attribute if it exists.

fields: QbFields = {'attributes': 'QbDictField(attributes.*) -> Dict[str, Any]',  'computer_pk': 'QbNumericField(attributes.computer_pk) -> Optional[int]',  'ctime': 'QbNumericField(ctime) -> datetime',  'description': 'QbStrField(description) -> str',  'detailed_job_info': 'QbDictField(attributes.detailed_job_info) -> '                       'Optional[dict]',  'exception': 'QbStrField(attributes.exception) -> Optional[str]',  'exit_message': 'QbStrField(attributes.exit_message) -> Optional[str]',  'exit_status': 'QbNumericField(attributes.exit_status) -> Optional[int]',  'extras': 'QbDictField(extras.*) -> Dict[str, Any]',  'imported': 'QbField(attributes.imported) -> Optional[bool]',  'job_id': 'QbStrField(attributes.job_id) -> Optional[str]',  'label': 'QbStrField(label) -> str',  'last_job_info': 'QbStrField(attributes.last_job_info) -> Optional[str]',  'mtime': 'QbNumericField(mtime) -> datetime',  'node_type': 'QbStrField(node_type) -> str',  'paused': 'QbField(attributes.paused) -> bool',  'pk': 'QbNumericField(pk) -> int',  'process_label': 'QbStrField(attributes.process_label) -> Optional[str]',  'process_state': 'QbStrField(attributes.process_state) -> Optional[str]',  'process_status': 'QbStrField(attributes.process_status) -> Optional[str]',  'process_type': 'QbStrField(attributes.process_type) -> Optional[str]',  'remote_workdir': 'QbStrField(attributes.remote_workdir) -> Optional[str]',  'repository_metadata': 'QbDictField(repository_metadata) -> Dict[str, Any]',  'retrieve_list': 'QbArrayField(attributes.retrieve_list) -> '                   'Optional[List[str]]',  'retrieve_temporary_list': 'QbArrayField(attributes.retrieve_temporary_list) '                             '-> Optional[List[str]]',  'scheduler_lastchecktime': 'QbStrField(attributes.scheduler_lastchecktime) -> '                             'Optional[str]',  'scheduler_state': 'QbStrField(attributes.scheduler_state) -> Optional[str]',  'sealed': 'QbField(attributes.sealed) -> bool',  'state': 'QbStrField(attributes.state) -> Optional[str]',  'user_pk': 'QbNumericField(user_pk) -> int',  'uuid': 'QbStrField(uuid) -> str'}

get_authinfo() → AuthInfo

Return the AuthInfo that is configured for the Computer set for this node.

返回:

AuthInfo

get_description() → str

Return a description of the node based on its properties.

get_detailed_job_info() → dict | None

Return the detailed job info dictionary.

The scheduler is polled for the detailed job info after the job is completed and ready to be retrieved.

返回:

the dictionary with detailed job info if defined or None

get_job_id() → str | None

Return job id that was assigned to the calculation by the scheduler.

返回:

the string representation of the scheduler job id

get_last_job_info() → JobInfo | None

Return the last information asked to the scheduler about the status of the job.

The last job info is updated on every poll of the scheduler, except for the final poll when the job drops from the scheduler’s job queue. For completed jobs, the last job info therefore contains the “second-to-last” job info that still shows the job as running. Please use get_detailed_job_info() instead.

返回:

a JobInfo object (that closely resembles a dictionary) or None.

get_option(name: str) → Any | None

Return the value of an option that was set for this CalcJobNode.

参数:

name – the option name

返回:

the option value or None

Raises:

ValueError for unknown option

get_options() → Dict[str, Any]

Return the dictionary of options set for this CalcJobNode

返回:

dictionary of the options and their values

get_parser_class() → Type[Parser] | None

Return the output parser object for this calculation or None if no parser is set.

返回:

a Parser class.

抛出:

aiida.common.exceptions.EntryPointError – if the parser entry point can not be resolved.

get_remote_workdir() → str | None

Return the path to the remote (on cluster) scratch folder of the calculation.

返回:

a string with the remote path

get_retrieve_list() → Sequence[str | Tuple[str, str, str]] | None

Return the list of files/directories to be retrieved on the cluster after the calculation has completed.

返回:

a list of file directives

get_retrieve_temporary_list() → Sequence[str | Tuple[str, str, str]] | None

Return list of files to be retrieved from the cluster which will be available during parsing.

返回:

a list of file directives

get_retrieved_node() → FolderData | None

Return the retrieved data folder.

返回:

the retrieved FolderData node or None if not found

get_scheduler_lastchecktime() → datetime | None

Return the time of the last update of the scheduler state by the daemon or None if it was never set.

返回:

a datetime object or None

get_scheduler_state() → JobState | None

Return the status of the calculation according to the cluster scheduler.

返回:

a JobState enum instance.

get_scheduler_stderr() → AnyStr | None

Return the scheduler stdout output if the calculation has finished and been retrieved, None otherwise.

返回:

scheduler stdout output or None

get_scheduler_stdout() → AnyStr | None

Return the scheduler stderr output if the calculation has finished and been retrieved, None otherwise.

返回:

scheduler stderr output or None

get_state() → CalcJobState | None

Return the calculation job active sub state.

The calculation job state serves to give more granular state information to CalcJobs, in addition to the generic process state, while the calculation job is active. The state can take values from the enumeration defined in aiida.common.datastructures.CalcJobState and can be used to query for calculation jobs in specific active states.

返回:

instance of aiida.common.datastructures.CalcJobState or None if invalid value, or not set

get_transport() → Transport

Return the transport for this calculation.

返回:

Transport configured with the AuthInfo associated to the computer of this node

property is_imported: bool

Return whether the calculation job was imported instead of being an actual run.

property link_label_retrieved: str

Return the link label used for the retrieved FolderData node.

property res: CalcJobResultManager

To be used to get direct access to the parsed parameters.

返回:

an instance of the CalcJobResultManager.

Note:

a practical example on how it is meant to be used: let’s say that there is a key ‘energy’ in the dictionary of the parsed results which contains a list of floats. The command calc.res.energy will return such a list.

set_detailed_job_info(detailed_job_info: dict | None) → None

Set the detailed job info dictionary.

参数:

detailed_job_info – a dictionary with metadata with the accounting of a completed job

set_job_id(job_id: int | str) → None

Set the job id that was assigned to the calculation by the scheduler.

备注

the id will always be stored as a string

参数:

job_id – the id assigned by the scheduler after submission

set_last_job_info(last_job_info: JobInfo) → None

Set the last job info.

参数:

last_job_info – a JobInfo object

set_option(name: str, value: Any) → None

Set an option to the given value

参数:

• name – the option name

• value – the value to set

Raises:

ValueError for unknown option

Raises:

TypeError for values with invalid type

set_options(options: Dict[str, Any]) → None

Set the options for this CalcJobNode

参数:

options – dictionary of option and their values to set

set_remote_workdir(remote_workdir: str) → None

Set the absolute path to the working directory on the remote computer where the calculation is run.

参数:

remote_workdir – absolute filepath to the remote working directory

set_retrieve_list(retrieve_list: Sequence[str | Tuple[str, str, str]]) → None

Set the retrieve list.

This list of directives will instruct the daemon what files to retrieve after the calculation has completed. list or tuple of files or paths that should be retrieved by the daemon.

参数:

retrieve_list – list or tuple of with filepath directives

set_retrieve_temporary_list(retrieve_temporary_list: Sequence[str | Tuple[str, str, str]]) → None

Set the retrieve temporary list.

The retrieve temporary list stores files that are retrieved after completion and made available during parsing and are deleted as soon as the parsing has been completed.

参数:

retrieve_temporary_list – list or tuple of with filepath directives

set_scheduler_state(state: JobState) → None

Set the scheduler state.

参数:

state – an instance of JobState

set_state(state: CalcJobState) → None

Set the calculation active job state.

Raise:

ValueError if state is invalid

property tools: CalculationTools

Return the calculation tools that are registered for the process type associated with this calculation.

If the entry point name stored in the process_type of the CalcJobNode has an accompanying entry point in the aiida.tools.calculations entry point category, it will attempt to load the entry point and instantiate it passing the node to the constructor. If the entry point does not exist, cannot be resolved or loaded, a warning will be logged and the base CalculationTools class will be instantiated and returned.

返回:

CalculationTools instance

class aiida.orm.nodes.process.calculation.calcjob.CalcJobNodeCaching(node: Node)

基类：ProcessNodeCaching

Interface to control caching of a node instance.

__annotations__ = {}

__module__ = 'aiida.orm.nodes.process.calculation.calcjob'

get_objects_to_hash() → List[Any]

Return a list of objects which should be included in the hash.

This method is purposefully overridden from the base Node class, because we do not want to include the repository folder in the hash. The reason is that the hash of this node is computed in the store method, at which point the input files that will be stored in the repository have not yet been generated. Including these anyway in the computation of the hash would mean that the hash of the node would change as soon as the process has started and the input files have been written to the repository.

Module with Node sub class for calculation processes.

class aiida.orm.nodes.process.calculation.calculation.CalculationNode(backend: StorageBackend | None = None, user: User | None = None, computer: Computer | None = None, **kwargs: Any)

基类：ProcessNode

Base class for all nodes representing the execution of a calculation process.

__abstractmethods__ = frozenset({})

__module__ = 'aiida.orm.nodes.process.calculation.calculation'

__parameters__ = ()

__plugin_type_string: ClassVar[str]

__query_type_string: ClassVar[str]

_abc_impl = <_abc._abc_data object>

_cachable = True

_logger: Logger | None = <Logger aiida.orm.nodes.process.calculation.calculation.CalculationNode (WARNING)>

_storable = True

_unstorable_message = 'storing for this node has been disabled'

fields: QbFields = {'attributes': 'QbDictField(attributes.*) -> Dict[str, Any]',  'computer_pk': 'QbNumericField(attributes.computer_pk) -> Optional[int]',  'ctime': 'QbNumericField(ctime) -> datetime',  'description': 'QbStrField(description) -> str',  'exception': 'QbStrField(attributes.exception) -> Optional[str]',  'exit_message': 'QbStrField(attributes.exit_message) -> Optional[str]',  'exit_status': 'QbNumericField(attributes.exit_status) -> Optional[int]',  'extras': 'QbDictField(extras.*) -> Dict[str, Any]',  'label': 'QbStrField(label) -> str',  'mtime': 'QbNumericField(mtime) -> datetime',  'node_type': 'QbStrField(node_type) -> str',  'paused': 'QbField(attributes.paused) -> bool',  'pk': 'QbNumericField(pk) -> int',  'process_label': 'QbStrField(attributes.process_label) -> Optional[str]',  'process_state': 'QbStrField(attributes.process_state) -> Optional[str]',  'process_status': 'QbStrField(attributes.process_status) -> Optional[str]',  'process_type': 'QbStrField(attributes.process_type) -> Optional[str]',  'repository_metadata': 'QbDictField(repository_metadata) -> Dict[str, Any]',  'sealed': 'QbField(attributes.sealed) -> bool',  'user_pk': 'QbNumericField(user_pk) -> int',  'uuid': 'QbStrField(uuid) -> str'}

property inputs: NodeLinksManager

Return an instance of NodeLinksManager to manage incoming INPUT_CALC links

The returned Manager allows you to easily explore the nodes connected to this node via an incoming INPUT_CALC link. The incoming nodes are reachable by their link labels which are attributes of the manager.

property outputs: NodeLinksManager

Return an instance of NodeLinksManager to manage outgoing CREATE links

The returned Manager allows you to easily explore the nodes connected to this node via an outgoing CREATE link. The outgoing nodes are reachable by their link labels which are attributes of the manager.