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

tensorflow · Dec 5, 2024 · 6e3a829 · 6e3a829
2 parents 53e530f + af06dc0
commit 6e3a829
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 46 deletions.
diff --git 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
@@ -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
@@ -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",
-]