import cloudpickle
import gin
import pickle


@gin.configurable
def msg_fn(msg):
    return msg


@gin.configurable
class Foo:
    def __init__(self, msg_fn):
        self.msg = msg_fn()


config = """
msg_fn.msg = "Hello World"
Foo.msg_fn = @msg_fn
"""
gin.parse_config(config)

try:
    pkl = pickle.dumps(Foo)
    foo = pickle.loads(pkl)()
    print("Can pickle: {}!".format(foo.msg))
except:
    print("Can't pickle!")

try:
    cpkl = cloudpickle.dumps(Foo)
    foo = cloudpickle.loads(cpkl)()
    print("Can cloudpickle: {}!".format(foo.msg))
except:
    print("Can't cloudpickle!")

Can pickle: Hello World!!
Can't cloudpickle!