@@ -233,7 +233,6 @@ def package(
233233 shutil .copyfile (file_name , main_path / f )
234234 create_file (output_path / "model.config.json" , srsly .json_dumps (meta , indent = 2 ))
235235 create_file (output_path / "setup.py" , TEMPLATE_SETUP )
236- create_file (output_path / "MANIFEST.in" , TEMPLATE_MANIFEST )
237236 create_file (package_path / "__init__.py" , TEMPLATE_INIT )
238237 msg .good ("Successfully created package '{}'" .format (package_path ), main_path )
239238 msg .text ("To build the package, run `python setup.py sdist` in this directory." )
@@ -248,7 +247,7 @@ def create_file(file_path: Path, contents: str):
248247TEMPLATE_SETUP = """
249248#!/usr/bin/env python
250249# coding: utf8
251- from __future__ import unicode_literals
250+ from typing import List
252251import io
253252import json
254253from os import path, walk
@@ -261,15 +260,17 @@ def load_meta(fp):
261260 return json.load(f)
262261
263262
264- def list_files(data_dir) :
263+ def list_files(data_dir: str, root: str) -> List[str] :
265264 output = []
266- for root, _, filenames in walk(data_dir):
265+ package_dir = path.join(root, data_dir)
266+ for folder, _, filenames in walk(package_dir):
267+ if "__pycache__" in folder:
268+ continue
267269 for filename in filenames:
268270 if not filename.startswith("."):
269- output.append(path.join(root, filename))
270- output = [path.relpath(p, path.dirname(data_dir)) for p in output]
271- output.append("model.config.json")
272- return output
271+ output.append(path.join(folder, filename))
272+ rel_output = [path.relpath(p, package_dir) for p in output]
273+ return rel_output
273274
274275
275276def list_requirements(meta):
@@ -287,9 +288,7 @@ def setup_package():
287288 meta_path = path.join(root, "model.config.json")
288289 meta = load_meta(meta_path)
289290 model_name = meta["name"]
290- model_dir = path.join(model_name, model_name))
291- copy(meta_path, path.join(model_name))
292- copy(meta_path, model_dir)
291+ model_dir = path.join(model_name)
293292 setup(
294293 name=model_name,
295294 description=meta["description"],
@@ -299,20 +298,14 @@ def setup_package():
299298 version=meta["version"],
300299 license=meta["license"],
301300 packages=[model_name],
302- package_data={model_name: list_files(model_dir)},
301+ package_data={model_name: list_files(model_dir, root )},
303302 install_requires=list_requirements(meta),
304303 zip_safe=False,
305304 )
306305
307306
308307if __name__ == "__main__":
309- setup_package()
310- """ .strip ()
311-
312-
313- TEMPLATE_MANIFEST = """
314- include model.config.json
315- """ .strip ()
308+ setup_package()""" .strip ()
316309
317310
318311TEMPLATE_INIT = """
0 commit comments