diff --git a/tfx/examples/penguin/penguin_pipeline_kubeflow.py b/tfx/examples/penguin/penguin_pipeline_kubeflow.py index 5a59b294bf..8c0e2e46ec 100644 --- a/tfx/examples/penguin/penguin_pipeline_kubeflow.py +++ b/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: diff --git a/tfx/v1/orchestration/experimental/__init__.py b/tfx/v1/orchestration/experimental/__init__.py index 43c5c89a31..7f2edb1b3b 100644 --- a/tfx/v1/orchestration/experimental/__init__.py +++ b/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 diff --git a/tfx/v1/proto/__init__.py b/tfx/v1/proto/__init__.py index 89a2f60b5c..30ca145ff1 100644 --- a/tfx/v1/proto/__init__.py +++ b/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", -]