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) selector and not selector(item)
): ):
continue continue
item = item if not projector else projector(item)
if return_obj: if return_obj:
item = tasks.ParamsItem() item = tasks.ParamsItem()
hyperparams[item.get("section")][item.get("name")] = ( hyperparams[item.get("section")][item.get("name")] = (

View File

@ -8,7 +8,6 @@ import threading
import time import time
from argparse import ArgumentParser from argparse import ArgumentParser
from logging import getLogger from logging import getLogger
from operator import attrgetter
from tempfile import mkstemp, mkdtemp from tempfile import mkstemp, mkdtemp
from zipfile import ZipFile, ZIP_DEFLATED from zipfile import ZipFile, ZIP_DEFLATED
@ -2180,7 +2179,7 @@ class Task(_Task):
section = "properties" section = "properties"
params = self._hyper_params_manager.get_hyper_params( 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, {})) return dict(params.get(section, {}))