Skip to content

Commit 1c2a429

Browse files
authored
fix setup.py (#2458)
1 parent c72d6c9 commit 1c2a429

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

setup.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,6 @@ def parse_requirements(fname):
3232

3333

3434
def package_model_zoo():
35-
from ppdet.model_zoo import MODEL_ZOO_FILENAME
3635
cur_dir = osp.dirname(osp.realpath(__file__))
3736
cfg_dir = osp.join(cur_dir, "configs")
3837
cfgs = glob.glob(osp.join(cfg_dir, '*/*.yml'))
@@ -44,7 +43,7 @@ def package_model_zoo():
4443
valid_cfgs.append(cfg)
4544
model_names = [osp.splitext(osp.split(cfg)[1])[0] for cfg in valid_cfgs]
4645

47-
model_zoo_file = osp.join(cur_dir, 'ppdet', 'model_zoo', MODEL_ZOO_FILENAME)
46+
model_zoo_file = osp.join(cur_dir, 'ppdet', 'model_zoo', 'MODEL_ZOO')
4847
with open(model_zoo_file, 'w') as wf:
4948
for model_name in model_names:
5049
wf.write("{}\n".format(model_name))

0 commit comments

Comments
 (0)