Exclude extractor_wrapper and generator_wrapper from the pip pack…

…age, and remove the exclusion for the deleted `create_python_api_wrapper`. PiperOrigin-RevId: 563777732
tensorflow · Sep 8, 2023 · bd50049 · bd50049
1 parent f7653f9
commit bd50049
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tensorflow_estimator/tools/pip_package/create_pip_helper.py b/tensorflow_estimator/tools/pip_package/create_pip_helper.py
@@ -35,7 +35,8 @@
     'tensorflow_estimator/python/estimator/canned/v1/baseline_test_v1.py',
     'tensorflow_estimator/python/estimator/canned/v1/dnn_test_fc_v1_v1.py',
     'tensorflow_estimator/python/estimator/canned/v1/dnn_test_fc_v2_v1.py',
-    'tensorflow_estimator/python/estimator/api/create_python_api_wrapper.py',
+    'tensorflow_estimator/python/estimator/api/extractor_wrapper.py',
+    'tensorflow_estimator/python/estimator/api/generator_wrapper.py',
     'tensorflow_estimator/tools/pip_package/setup.py',
     'tensorflow_estimator/tools/pip_package/create_pip_helper.py',
 ])