from aim.artifacts.map import *
from aim.artifacts.metric import *
from aim.artifacts.model import *

# Serializable classes references
metric = 'Metric'
hyperparams = 'HyperParameters'
dataset = 'Dataset'
map = 'Map'
nested_map = 'NestedMap'
checkpoint = 'Checkpoint'

# Serializable aliases
params = hyperparams
dictionary = nested_map
