make Dataset* as an argument

6 years ago · 71aa307ebe
parent b415ec27e8
commit 71aa307ebe
1 changed files with 1 additions and 0 deletions
--- a/python/paddle/fluid/distributed/fleet.py
+++ b/python/paddle/fluid/distributed/fleet.py
@ -11,6 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 from .. import core
+from . import ps_instance

 __all__ = ['Fleet']