update setup.py

6f7d5975 · tangwei · 31bbef3d · 6f7d5975
隐藏空白更改
内联并排

Showing with 2 addition and 3 deletion

setup.py setup.py +2 -3

未找到文件。
--- a/setup.py
+++ b/setup.py
@@ -4,9 +4,8 @@ setup for fleet-rec.

 from setuptools import setup

-packages = ["fleetrec", "fleetrec.examples", "fleetrec.metrics", "fleetrec.models", "fleetrec.models.ctr_dnn",
-            "fleetrec.reader",
-            "fleetrec.trainer", "fleetrec.utils"]
+packages = ["fleetrec", "fleetrec.models", "fleetrec.examples", "fleetrec.core", "fleetrec.core.engine",
+            "fleetrec.core.metrics", "fleetrec.core.models", "fleetrec.core.trainers", "fleetrec.core.utils"]

 requires = [
    "paddlepaddle"