Merge branch '1.16' of https://github.com/nikelite/tfx into 1.16

Author: nikelite

tfx/examples/penguin/penguin_pipeline_kubeflow.py

@@ -503,8 +503,8 @@ def main():
 
   dag_runner = tfx.orchestration.experimental.KubeflowV2DagRunner(
       config=tfx.orchestration.experimental.KubeflowV2DagRunnerConfig(),
-      output_filename=_pipeline_definition_file)
-
+      output_filename=_pipeline_definition_file,
+  )
   dag_runner.run(
       create_pipeline(
           pipeline_name=_pipeline_name,
@@ -521,7 +521,8 @@ def main():
           use_aip=use_aip,
           use_vertex=use_vertex,
           serving_model_dir=_serving_model_dir,
-      ))
+      )
+  )
 
 
 # To compile the pipeline:

tfx/v1/orchestration/experimental/__init__.py

@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """TFX orchestration.experimental module."""
-
 try:
     from tfx.orchestration.kubeflow.decorators import exit_handler  # pylint: disable=g-import-not-at-top
     from tfx.orchestration.kubeflow.decorators import FinalStatusStr  # pylint: disable=g-import-not-at-top

tfx/v1/proto/__init__.py

@@ -233,45 +233,3 @@
 PairedExampleSkew.__doc__ = """
 Configurations related to Example Diff on feature pairing level.
 """
-
-__all__ = [
-    "orchestration",
-    "ClassifyOutput",
-    "CustomConfig",
-    "DataSpec",
-    "DistributionValidatorConfig",
-    "EnvVar",
-    "EnvVarSource",
-    "EvalArgs",
-    "ExampleDiffConfig",
-    "FeatureComparator",
-    "FeatureSlicingSpec",
-    "Filesystem",
-    "Input",
-    "KubernetesConfig",
-    "LocalDockerConfig",
-    "ModelSpec",
-    "Output",
-    "OutputColumnsSpec",
-    "OutputExampleSpec",
-    "PairedExampleSkew",
-    "PodOverrides",
-    "PredictOutput",
-    "PredictOutputCol",
-    "PushDestination",
-    "RangeConfig",
-    "RegressOutput",
-    "RequestSpec",
-    "RollingRange",
-    "SecretKeySelector",
-    "ServingSpec",
-    "SingleSlicingSpec",
-    "SplitConfig",
-    "SplitsConfig",
-    "StaticRange",
-    "TensorFlowServing",
-    "TensorFlowServingRequestSpec",
-    "TrainArgs",
-    "TuneArgs",
-    "ValidationSpec",
-]