Fix ruff rules

janasangeetha · janasangeetha · commit c034df5104a5 · 2024-11-06T09:48:54.000+05:30
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:# noqa: B027
+  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
@@ -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):# noqa: B024
+class SystemArtifact(abc.ABC):  # noqa: B024
   """TFX system artifact base class.
 
   A user may create a subclass of SystemArtifact and override the
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):# noqa: B024
+class SystemExecution(abc.ABC):  # noqa: B024
   """TFX system execution base class.
 
   A user may create a subclass of SystemExecution and override the
