From 701186d8a8f73d5df4c8dfc21720f0b6a09e04e4 Mon Sep 17 00:00:00 2001 From: subhash kumar singh Date: Thu, 5 Dec 2024 10:29:22 +0000 Subject: [PATCH] Fix create feature gruop Return feature group on creation. Also, include the pip package in tox. Change-Id: I30787851817ba8517a798cda389b8dcafeaa32f2 Signed-off-by: subhash kumar singh --- tox.ini | 6 +----- trainingmgr/trainingmgr_main.py | 2 +- 2 files changed, 2 insertions(+), 6 deletions(-) diff --git a/tox.ini b/tox.ini index 92d3a48..e3606f1 100644 --- a/tox.ini +++ b/tox.ini @@ -51,11 +51,7 @@ setenv = cd = {toxinidir}/tests commands = pip3 install -e {toxinidir} - git clone "https://gerrit.o-ran-sc.org/r/aiml-fw/athp/sdk/feature-store" /tmp/fssdk/ - git clone "https://gerrit.o-ran-sc.org/r/aiml-fw/athp/sdk/model-storage" /tmp/modelsdk/ - - pip3 install /tmp/fssdk/. - pip3 install /tmp/modelsdk/. + pip3 install featurestoresdk modelmetricsdk pytest --cov {toxinidir}/trainingmgr --cov-report xml --cov-report term-missing --cov-report html --cov-fail-under=10 --junitxml=/tmp/tests.xml coverage xml -i diff --git a/trainingmgr/trainingmgr_main.py b/trainingmgr/trainingmgr_main.py index a38bfbf..aad71e3 100644 --- a/trainingmgr/trainingmgr_main.py +++ b/trainingmgr/trainingmgr_main.py @@ -1244,7 +1244,7 @@ def create_feature_group(): else: # the features are stored in string format in the db, and has to be passed as list of feature to the dme. Hence the conversion. add_featuregroup(featuregroup) - api_response={"result": "Feature Group Created"} + api_response = FeatureGroupSchema().dump(featuregroup) response_code =status.HTTP_200_OK if featuregroup.enable_dme == True : response= create_dme_filtered_data_job(TRAININGMGR_CONFIG_OBJ, featuregroup) -- 2.16.6