Without this import the example in DataSaver class doesn't work.

from collections import OrderedDict

import functools

import operator

class DataSaver:

    """Save extra data associated with the values that need to be learned.