From 8cd18a23907384289ff399eebf5be9de17a30ee3 Mon Sep 17 00:00:00 2001 From: neeratyoy Date: Tue, 16 Jul 2019 15:42:06 +0200 Subject: [PATCH] Adding __repr__ --- openml/datasets/dataset.py | 3 +++ openml/evaluations/evaluation.py | 3 +++ openml/flows/flow.py | 3 +++ openml/runs/run.py | 3 +++ openml/runs/trace.py | 3 +++ openml/setups/setup.py | 6 ++++++ openml/study/study.py | 6 ++++++ openml/tasks/task.py | 3 +++ 8 files changed, 30 insertions(+) diff --git a/openml/datasets/dataset.py b/openml/datasets/dataset.py index 6cf0b3a31..5e3f453ab 100644 --- a/openml/datasets/dataset.py +++ b/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): diff --git a/openml/evaluations/evaluation.py b/openml/evaluations/evaluation.py index f22ec36cf..98c0ba74a 100644 --- a/openml/evaluations/evaluation.py +++ b/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__() diff --git a/openml/flows/flow.py b/openml/flows/flow.py index bdd4fe6a6..34320f671 100644 --- a/openml/flows/flow.py +++ b/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. diff --git a/openml/runs/run.py b/openml/runs/run.py index 65c525115..52f12140f 100644 --- a/openml/runs/run.py +++ b/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)) diff --git a/openml/runs/trace.py b/openml/runs/trace.py index 42e89c50b..bc33b09d5 100644 --- a/openml/runs/trace.py +++ b/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 diff --git a/openml/setups/setup.py b/openml/setups/setup.py index 6c4a240c1..1325f3f06 100644 --- a/openml/setups/setup.py +++ b/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__() diff --git a/openml/study/study.py b/openml/study/study.py index 259453422..5549fb936 100644 --- a/openml/study/study.py +++ b/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__() diff --git a/openml/tasks/task.py b/openml/tasks/task.py index 831825592..5937f51bf 100644 --- a/openml/tasks/task.py +++ b/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)