Skip to content

Commit 8317ada

Browse files
author
arnaudde
committed
Add two missing reprs and fix one
1 parent e020979 commit 8317ada

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

dataikuapi/dss/ml.py

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -663,6 +663,9 @@ class DSSMLAssertionsParams(object):
663663
def __init__(self, data):
664664
self._internal_dict = data
665665

666+
def __repr__(self):
667+
return u"{}({})".format(self.__class__.__name__, self.get_raw())
668+
666669
def get_raw(self):
667670
"""
668671
Gets the raw dictionary of the assertions parameters
@@ -920,6 +923,9 @@ class DSSMLAssertionsMetrics(object):
920923
def __init__(self, data):
921924
self._internal_dict = data
922925

926+
def __repr__(self):
927+
return u"{}({})".format(self.__class__.__name__, self.get_raw())
928+
923929
def get_raw(self):
924930
"""
925931
Gets the raw dictionary of the assertions metrics
@@ -962,9 +968,10 @@ def __init__(self, data):
962968
self._internal_dict = data
963969

964970
def __repr__(self):
965-
return u"DSSMLAssertionParams(name='{}', result={}, valid_ratio={}, nb_matching_rows={}," \
966-
u" nb_dropped_rows={})".format(self.name, self.result, self.valid_ratio, self.nb_matching_rows,
967-
self.nb_dropped_rows)
971+
return u"{}(name='{}', result={}, valid_ratio={}, nb_matching_rows={}," \
972+
u" nb_dropped_rows={})".format(self.__class__.__name__, self.name, self.result, self.valid_ratio,
973+
self.nb_matching_rows, self.nb_dropped_rows)
974+
968975
def get_raw(self):
969976
"""
970977
Gets the raw dictionary of metrics of one assertion

0 commit comments

Comments
 (0)