Move server_lib under tensorflow/python to fix nightly pip build.

tf_python_pybind_extension requires that it be used under the tensorflow/python directory.

PiperOrigin-RevId: 304490055
Change-Id: If9e16aa20e4223149f1d1aa7d732db7bae033461

tensorflow/python/data/service/server_lib.py

@@ -20,7 +20,7 @@ from __future__ import print_function
 
 # pylint: disable=invalid-import-order,g-bad-import-order, unused-import
 from tensorflow.python import pywrap_tensorflow
-from tensorflow.core.data.service.python import _pywrap_server_lib
+from tensorflow.python.data.service import _pywrap_server_lib
 
 
 class MasterServer(object):

tensorflow/python/data/service/server_lib_test.py

@@ -18,7 +18,7 @@ from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
 
-from tensorflow.core.data.service.python import server_lib
+from tensorflow.python.data.service import server_lib
 
 from tensorflow.python.platform import test
 

tensorflow/tools/pip_package/BUILD

@@ -95,7 +95,6 @@ COMMON_PIP_DEPS = [
     "//tensorflow:tensorflow_py",
     "//tensorflow/compiler/tf2xla:xla_compiled_cpu_runtime_hdrs",
     "//tensorflow/compiler/tf2xla:xla_compiled_cpu_runtime_srcs",
-    "//tensorflow/core/data/service/python:server_lib",
     "//tensorflow/core:protos_all_proto_srcs",
     "//tensorflow/examples/saved_model/integration_tests:mnist_util",
     "//tensorflow/lite/python/testdata:interpreter_test_data",
@@ -107,6 +106,7 @@ COMMON_PIP_DEPS = [
     "//tensorflow/python/autograph/pyct/common_transformers:common_transformers",
     "//tensorflow/python/compiler:compiler",
     "//tensorflow/python:cond_v2",
+    "//tensorflow/python/data/service:server_lib",
     "//tensorflow/python:distributed_framework_test_lib",
     "//tensorflow/python/distribute:distribute_test_lib_pip",
     "//tensorflow/python:loss_scale",