Skip to content

Commit 75a755e

Browse files
author
arnaudde
committed
Add get_assertions_names
1 parent b10aa6f commit 75a755e

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

dataikuapi/dss/ml.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -687,6 +687,15 @@ def get_assertion(self, assertion_name):
687687
return DSSMLAssertionParams(assertion_dict)
688688
return None
689689

690+
def get_assertions_names(self):
691+
"""
692+
Gets the list of all assertion's names
693+
694+
:return: A list of all assertion's names
695+
:rtype: list
696+
"""
697+
return [assertion_dict["name"] for assertion_dict in self._internal_dict["assertions"]]
698+
690699
def add_assertion(self, assertion_params):
691700
"""
692701
Adds parameters of an assertion to the assertions parameters of the ml task.

0 commit comments

Comments
 (0)