Merge pull request #6950 from janasangeetha/sangeethajana-patch1

lego0901 · web-flow · commit 8393d144518e · 2024-11-15T13:28:28.000+09:00
Fix Ruff rule B027 and B024
diff --git a/tfx/orchestration/portable/base_executor_operator.py b/tfx/orchestration/portable/base_executor_operator.py
@@ -84,6 +84,6 @@ def with_execution_watcher(
     self._execution_watcher_address = execution_watcher_address
     return self
 
-  def handle_stop(self) -> None:
+  def handle_stop(self) -> None:  # noqa: B027
     """Executor Operator specific logic to clean up after it is stopped."""
     pass
diff --git a/tfx/tools/cli/handler/dag_runner_patcher.py b/tfx/tools/cli/handler/dag_runner_patcher.py
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Base class to patch DagRunner classes in TFX CLI."""
-
+#ruff: noqa: B027
 import abc
 import contextlib
 import functools
@@ -56,12 +56,12 @@ def __init__(self, call_real_run=True):
     self._run_called = False
     self._call_real_run = call_real_run
 
-  def _before_run(self, runner: tfx_runner.TfxRunner,
+  def _before_run(self, runner: tfx_runner.TfxRunner,  # noqa: B027
                   pipeline: Union[pipeline_pb2.Pipeline, tfx_pipeline.Pipeline],
                   context: MutableMapping[str, Any]) -> None:
     pass
 
-  def _after_run(self, runner: tfx_runner.TfxRunner,
+  def _after_run(self, runner: tfx_runner.TfxRunner,  # noqa: B027
                  pipeline: Union[pipeline_pb2.Pipeline, tfx_pipeline.Pipeline],
                  context: MutableMapping[str, Any]) -> None:
     pass
diff --git a/tfx/types/system_artifacts.py b/tfx/types/system_artifacts.py
@@ -21,7 +21,7 @@
 from ml_metadata.metadata_store import mlmd_types
 
 
-class SystemArtifact(abc.ABC):
+class SystemArtifact(abc.ABC):  # noqa: B024
   """TFX system artifact base class.
 
   A user may create a subclass of SystemArtifact and override the
@@ -30,7 +30,6 @@ class SystemArtifact(abc.ABC):
   The subclasses, e.g, Dataset, Model, Statistics, e.t.c, match the MLMD types
   from third_party/ml_metadata/metadata_store/mlmd_types.py.
   """
-
   # MLMD system base type enum. Override it when creating subclasses.
   MLMD_SYSTEM_BASE_TYPE = None
 
diff --git a/tfx/types/system_executions.py b/tfx/types/system_executions.py
@@ -21,7 +21,7 @@
 from ml_metadata.metadata_store import mlmd_types
 
 
-class SystemExecution(abc.ABC):
+class SystemExecution(abc.ABC):  # noqa: B024
   """TFX system execution base class.
 
   A user may create a subclass of SystemExecution and override the
@@ -30,7 +30,6 @@ class SystemExecution(abc.ABC):
   The subclasses, e.g, Train, Transform, Process, e.t.c, match the MLMD types
   from third_party/ml_metadata/metadata_store/mlmd_types.py.
   """
-
   # MLMD system base type enum. Override it when creating subclasses.
   MLMD_SYSTEM_BASE_TYPE = None
 
