fix: export directory issue

This commit is contained in:
Félix Voituret
2019-11-06 20:36:31 +01:00
parent 01319e1a63
commit 2559d52ecf

View File

@@ -4,6 +4,8 @@
""" Utility functions for creating estimator. """
from pathlib import Path
from os.path import join
from tempfile import gettempdir
# pylint: disable=import-error
import tensorflow as tf
@@ -15,7 +17,7 @@ from ..model import model_fn
from ..model.provider import get_default_model_provider
# Default exporting directory for predictor.
DEFAULT_EXPORT_DIRECTORY = '/tmp/serving'
DEFAULT_EXPORT_DIRECTORY = join(gettempdir(), 'serving')
def create_estimator(params, MWF):