fix (#2154)

intel-analytics · Jan 4, 2018 · 4188a40 · 4188a40
1 parent eb5d1a1
commit 4188a40
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pyspark/setup.py b/pyspark/setup.py
@@ -81,9 +81,16 @@ def setup_package():
                   'bigdl.dataset',
                   'bigdl.nn',
                   'bigdl.transform',
+                  'bigdl.transform.vision',
                   'bigdl.keras',
                   'bigdl.examples',
+                  'bigdl.examples.keras',
                   'bigdl.models',
+                  'bigdl.models.lenet',
+                  'bigdl.models.local_lenet',
+                  'bigdl.models.ml_pipeline',
+                  'bigdl.models.rnn',
+                  'bigdl.models.textclassifier',
                   'bigdl.optim',
                   'bigdl.util',
                   'bigdl.share'],