Fix task.get_user_properties when called with value_only=True

This commit is contained in:
Alex Burlacu 2023-03-23 13:16:06 +02:00
parent 2e6e0c1a1d
commit 90354a7751
2 changed files with 1 additions and 3 deletions

View File

@ -42,7 +42,6 @@ class HyperParams(object):
selector and not selector(item)
):
continue
item = item if not projector else projector(item)
if return_obj:
item = tasks.ParamsItem()
hyperparams[item.get("section")][item.get("name")] = (

View File

@ -8,7 +8,6 @@ import threading
import time
from argparse import ArgumentParser
from logging import getLogger
from operator import attrgetter
from tempfile import mkstemp, mkdtemp
from zipfile import ZipFile, ZIP_DEFLATED
@ -2180,7 +2179,7 @@ class Task(_Task):
section = "properties"
params = self._hyper_params_manager.get_hyper_params(
sections=[section], projector=attrgetter("value") if value_only else None
sections=[section], projector=(lambda x: x.get("value")) if value_only else None
)
return dict(params.get(section, {}))