From 10d13c434159cc917a2e752f24d5f4c745cc88a3 Mon Sep 17 00:00:00 2001 From: Kaipeng Deng Date: Tue, 30 Mar 2021 14:11:29 +0800 Subject: [PATCH] fix setup.py (#2457) --- setup.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 5c2f365b3..7be74fce9 100644 --- a/setup.py +++ b/setup.py @@ -32,7 +32,6 @@ def parse_requirements(fname): def package_model_zoo(): - from ppdet.model_zoo import MODEL_ZOO_FILENAME cur_dir = osp.dirname(osp.realpath(__file__)) cfg_dir = osp.join(cur_dir, "configs") cfgs = glob.glob(osp.join(cfg_dir, '*/*.yml')) @@ -44,7 +43,7 @@ def package_model_zoo(): valid_cfgs.append(cfg) model_names = [osp.splitext(osp.split(cfg)[1])[0] for cfg in valid_cfgs] - model_zoo_file = osp.join(cur_dir, 'ppdet', 'model_zoo', MODEL_ZOO_FILENAME) + model_zoo_file = osp.join(cur_dir, 'ppdet', 'model_zoo', 'MODEL_ZOO') with open(model_zoo_file, 'w') as wf: for model_name in model_names: wf.write("{}\n".format(model_name)) -- GitLab