Adding __repr__ to top level classes

openml/datasets/dataset.py

@@ -203,6 +203,9 @@ def __str__(self):
         body = '\n'.join(field_line_format.format(name, value) for name, value in fields)
         return header + body
 
+    def __repr__(self):
+        return self.__str__()
+
     def _data_arff_to_pickle(self, data_file):
         data_pickle_file = data_file.replace('.arff', '.pkl.py3')
         if os.path.exists(data_pickle_file):

openml/evaluations/evaluation.py

@@ -78,3 +78,6 @@ def __str__(self):
         field_line_format = "{{:.<{}}}: {{}}".format(longest_field_name_length)
         body = '\n'.join(field_line_format.format(name, value) for name, value in fields)
         return header + body
+
+    def __repr__(self):
+        return self.__str__()

openml/flows/flow.py

@@ -171,6 +171,9 @@ def __str__(self):
         body = '\n'.join(field_line_format.format(name, value) for name, value in fields)
         return header + body
 
+    def __repr__(self):
+        return self.__str__()
+
     def _to_xml(self) -> str:
         """Generate xml representation of self for upload to server.
 

openml/runs/run.py

@@ -103,6 +103,9 @@ def __str__(self):
         body = '\n'.join(field_line_format.format(name, value) for name, value in fields)
         return header + body
 
+    def __repr__(self):
+        return self.__str__()
+
     def _repr_pretty_(self, pp, cycle):
         pp.text(str(self))
 

openml/runs/trace.py

@@ -386,6 +386,9 @@ def __str__(self):
             len(self.trace_iterations),
         )
 
+    def __repr__(self):
+        return self.__str__()
+
     def __iter__(self):
         for val in self.trace_iterations.values():
             yield val

openml/setups/setup.py

@@ -46,6 +46,9 @@ def __str__(self):
         body = '\n'.join(field_line_format.format(name, value) for name, value in fields)
         return header + body
 
+    def __repr__(self):
+        return self.__str__()
+
 
 class OpenMLParameter(object):
     """Parameter object (used in setup).
@@ -112,3 +115,6 @@ def __str__(self):
         field_line_format = "{{:.<{}}}: {{}}".format(longest_field_name_length)
         body = '\n'.join(field_line_format.format(name, value) for name, value in fields)
         return header + body
+
+    def __repr__(self):
+        return self.__str__()

openml/study/study.py

@@ -274,6 +274,9 @@ def __str__(self):
         body = super(OpenMLStudy, self).__str__()
         return header + body
 
+    def __repr__(self):
+        return self.__str__()
+
 
 class OpenMLBenchmarkSuite(BaseStudy):
     """
@@ -351,3 +354,6 @@ def __str__(self):
         header = '{}\n{}\n'.format(header, '=' * len(header))
         body = super(OpenMLBenchmarkSuite, self).__str__()
         return header + body
+
+    def __repr__(self):
+        return self.__str__()

openml/tasks/task.py

@@ -85,6 +85,9 @@ def __str__(self):
         body = '\n'.join(field_line_format.format(name, value) for name, value in fields)
         return header + body
 
+    def __repr__(self):
+        return self.__str__()
+
     def get_dataset(self) -> datasets.OpenMLDataset:
         """Download dataset associated with task"""
         return datasets.get_dataset(self.dataset_id)