Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add to_aiida_type to the public API #4672

Merged
merged 9 commits into from
Jan 27, 2021
4 changes: 2 additions & 2 deletions aiida/orm/nodes/data/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
"""Module with `Node` sub classes for data structures."""

from .array import ArrayData, BandsData, KpointsData, ProjectionData, TrajectoryData, XyData
from .base import BaseType
from .base import BaseType, to_aiida_type
from .bool import Bool
from .cif import CifData
from .code import Code
Expand All @@ -31,5 +31,5 @@
__all__ = (
'Data', 'BaseType', 'ArrayData', 'BandsData', 'KpointsData', 'ProjectionData', 'TrajectoryData', 'XyData', 'Bool',
'CifData', 'Code', 'Float', 'FolderData', 'Int', 'List', 'OrbitalData', 'Dict', 'RemoteData', 'SinglefileData',
'Str', 'StructureData', 'UpfData', 'NumericType'
'Str', 'StructureData', 'UpfData', 'NumericType', 'to_aiida_type'
)
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
# -*- coding: utf-8 -*-
from aiida.engine import WorkChain
from aiida.orm.nodes.data import to_aiida_type
from aiida.orm import to_aiida_type
# The basic types need to be loaded such that they are registered with
# the 'to_aiida_type' function.
from aiida.orm.nodes.data.base import *
chrisjsewell marked this conversation as resolved.
Show resolved Hide resolved
chrisjsewell marked this conversation as resolved.
Show resolved Hide resolved
Expand Down