Update global config
# -*- coding: utf-8 -*-
from __future__ import absolute_import, division, print_function
-from marshmallow import Schema, fields, post_load
+from marshmallow import Schema, fields, post_load, post_dump
from polyaxon_schemas.base import BaseConfig
@@ -18,6 +18,10 @@ class GlobalConfigurationSchema(Schema):
def make(self, data):
return GlobalConfigurationConfig(**data)
+ @post_dump
+ def unmake(self, data):
+ return GlobalConfigurationConfig.remove_reduced_attrs(data)
+
class GlobalConfigurationConfig(BaseConfig):
SCHEMA = GlobalConfigurationSchema
