From 8f4e6e9c3cf40b61478ccebff151d0f010713258 Mon Sep 17 00:00:00 2001 From: "jordi.hasianta" Date: Thu, 15 Sep 2022 20:22:24 +0700 Subject: [PATCH] import dbm_lib_derive_feature_config directly --- .../dbm_lib/config/config_derive_feature.py | 20 ++++++++++++------- 1 file changed, 13 insertions(+), 7 deletions(-) diff --git a/opendbm/dbm_lib/config/config_derive_feature.py b/opendbm/dbm_lib/config/config_derive_feature.py index 73c6af16..5562bf41 100644 --- a/opendbm/dbm_lib/config/config_derive_feature.py +++ b/opendbm/dbm_lib/config/config_derive_feature.py @@ -4,26 +4,32 @@ project_name: DBM created: 2020-20-07 """ +import os + import yaml -from opendbm.dbm_lib import DBMLIB_DERIVE_FEATURE_CONFIG + +DBMLIB_PATH = os.path.dirname(__file__) +DBMLIB_DERIVE_FEATURE_CONFIG = os.path.abspath( + os.path.join(DBMLIB_PATH, "../../resources/features/derived_feature.yml") +) + class ConfigDeriveReader(object): """Summary Read sevice end ponit """ - def __init__(self, - feature_config_yml=None): + + def __init__(self, feature_config_yml=None): """Summary Args: feature_config_yml (None, optional): yml file defined service configuration """ - + if feature_config_yml is None: feature_config = DBMLIB_DERIVE_FEATURE_CONFIG else: feature_config = feature_config_yml - with open(feature_config, 'r') as ymlfile: - config = yaml.load(ymlfile) + with open(feature_config, "r") as ymlfile: + config = yaml.load(ymlfile, Loader=yaml.CLoader) self.base_derive = config - \ No newline at end of file