From 6f1c208726448316ecd8c498e3844fc599bb2075 Mon Sep 17 00:00:00 2001 From: Harry Date: Tue, 28 Mar 2023 10:21:43 +0100 Subject: [PATCH] Add a cast parameter to `Task.get_parameter()` (#958) --- clearml/backend_interface/task/task.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/clearml/backend_interface/task/task.py b/clearml/backend_interface/task/task.py index f32ee1fa..31aeb6ee 100644 --- a/clearml/backend_interface/task/task.py +++ b/clearml/backend_interface/task/task.py @@ -1320,16 +1320,17 @@ class Task(IdObjectBase, AccessMixin, SetupUploadMixin): __parameters_types={name: value_type} ) - def get_parameter(self, name, default=None): - # type: (str, Any) -> Any + def get_parameter(self, name, default=None, cast=False): + # type: (str, Any, bool) -> Any """ Get a value for a parameter. :param name: Parameter name :param default: Default value + :param cast: If value is found, cast to original type. If False, return string. :return: The Parameter value (or default value if parameter is not defined). """ - params = self.get_parameters() + params = self.get_parameters(cast=cast) return params.get(name, default) def delete_parameter(self, name):