Replace access to internal dict by get raw

Author: arnaudde

dataikuapi/dss/ml.py

@@ -693,7 +693,7 @@ def add_assertion(self, assertion_params):
         if not isinstance(assertion_params, DSSMLAssertionParams):
             raise ValueError('Wrong type for assertion parameters: {}'.format(type(assertion_params)))
 
-        self._internal_dict["assertions"].append(assertion_params._internal_dict)
+        self._internal_dict["assertions"].append(assertion_params.get_raw())
 
     def delete_assertion(self, assertion_name):
         """
@@ -786,7 +786,7 @@ def condition(self):
     def condition(self, condition):
         if not isinstance(condition, DSSMLAssertionCondition):
             raise ValueError('Wrong type for assertion condition: {}'.format(type(condition)))
-        self._internal_dict["assertionCondition"] = condition._internal_dict
+        self._internal_dict["assertionCondition"] = condition.get_raw()
 
 
 class DSSMLAssertionCondition(object):