Fix Ruff rule A002

Author: smokestacklightnin

Diff for: tfx/dsl/compiler/node_inputs_compiler_test.py

@@ -47,7 +47,7 @@ class DummyArtifact(types.Artifact):
 
 class DummyNode(base_node.BaseNode):
 
-  def __init__(self, id: str, inputs=None, exec_properties=None):  # pylint: disable=redefined-builtin
+  def __init__(self, id: str, inputs=None, exec_properties=None):  # noqa: A002
     super().__init__()
     self.with_id(id)
     self._inputs = inputs or {}

Diff for: tfx/dsl/components/base/base_component_test.py

@@ -61,7 +61,7 @@ class _BasicComponent(base_component.BaseComponent):
   def __init__(self,
                spec: types.ComponentSpec = None,
                folds: int = None,
-               input: types.Channel = None):  # pylint: disable=redefined-builtin
+               input: types.Channel = None):  # noqa: A002
     if not spec:
       output = types.Channel(type=_OutputArtifact)
       spec = _BasicComponentSpec(folds=folds, input=input, output=output)

Diff for: tfx/dsl/components/base/base_node.py

@@ -126,12 +126,12 @@ def component_id(self) -> str:
 
   @id.setter
   @doc_controls.do_not_doc_in_subclasses
-  def id(self, id: str) -> None:  # pylint: disable=redefined-builtin
+  def id(self, id: str) -> None:  # noqa: A002
     self._id = id
 
   # TODO(kmonte): Update this to Self once we're on 3.11 everywhere
   @doc_controls.do_not_doc_in_subclasses
-  def with_id(self, id: str) -> typing_extensions.Self:  # pylint: disable=redefined-builtin
+  def with_id(self, id: str) -> typing_extensions.Self:  # noqa: A002
     self._id = id
     return self
 

Diff for: tfx/dsl/input_resolution/canned_resolver_functions.py

@@ -744,7 +744,7 @@ def pick(channel: channel_types.BaseChannel, i: int, /):
   return _slice(channel, start=i, stop=(i + 1) or None, min_count=1)
 
 
-def slice(  # pylint: disable=redefined-builtin
+def slice(  # noqa: A002
     channel: channel_types.BaseChannel,
     /,
     start: Optional[int] = None,

Diff for: tfx/dsl/input_resolution/ops/test_utils.py

@@ -51,7 +51,7 @@ class DummyArtifact(types.Artifact):
       'version': tfx_artifact.Property(type=tfx_artifact.PropertyType.INT),
   }
 
-  # pylint: disable=redefined-builtin
+  # noqa: A002
   def __init__(
       self,
       id: Optional[str] = None,
@@ -149,7 +149,7 @@ class FakeComponent(base_component.BaseComponent):
 
   DRIVER_CLASS = base_driver.BaseDriver
 
-  def __init__(self, id: str, inputs=None, exec_properties=None):  # pylint: disable=redefined-builtin
+  def __init__(self, id: str, inputs=None, exec_properties=None):  # noqa: A002
     super().__init__(spec=FakeSpec())
     self.with_id(id)
 

Diff for: tfx/dsl/input_resolution/resolver_op.py

@@ -228,7 +228,7 @@ def my_resolver_stem(input_dict):
 
   def __init__(
       self, *,
-      type: Type[_T],  # pylint: disable=redefined-builtin
+      type: Type[_T],  # noqa: A002
       default: Union[_T, _Empty] = _EMPTY):
     self._type = type
     self._required = default is _EMPTY

Diff for: tfx/dsl/io/fileio.py

@@ -20,8 +20,8 @@
 from tfx.dsl.io.filesystem import PathType
 
 # Import modules that may provide filesystem plugins.
-import tfx.dsl.io.plugins.tensorflow_gfile  # pylint: disable=unused-import, g-import-not-at-top
-import tfx.dsl.io.plugins.local  # pylint: disable=unused-import, g-import-not-at-top
+import tfx.dsl.io.plugins.tensorflow_gfile  # noqa: F401, E402
+import tfx.dsl.io.plugins.local  # noqa: F401, E402
 
 
 # Expose `NotFoundError` as `fileio.NotFoundError`.
@@ -33,7 +33,7 @@ def _get_filesystem(path) -> Type[filesystem.Filesystem]:
           .get_filesystem_for_path(path))
 
 
-def open(path: PathType, mode: str = 'r'):  # pylint: disable=redefined-builtin
+def open(path: PathType, mode: str = 'r'):  # noqa: A002
   """Open a file at the given path."""
   return _get_filesystem(path).open(path, mode=mode)
 

Diff for: tfx/dsl/placeholder/artifact_placeholder.py

@@ -23,7 +23,7 @@
 _types = placeholder_base.types
 
 
-def input(key: str) -> ArtifactPlaceholder:  # pylint: disable=redefined-builtin
+def input(key: str) -> ArtifactPlaceholder:  # noqa: A002
   """Returns a Placeholder that represents an input artifact.
 
   Args:

Diff for: tfx/orchestration/pipeline_test.py

@@ -85,7 +85,7 @@ class _FakeComponent(base_component.BaseComponent):
 
     def __init__(
         self,
-        type: Type[types.Artifact],  # pylint: disable=redefined-builtin
+        type: Type[types.Artifact],  # noqa: A002
         spec_kwargs: Dict[str, Any]):
       spec = _FakeComponentSpec(output=types.Channel(type=type), **spec_kwargs)
       super().__init__(spec=spec)
@@ -98,7 +98,7 @@ class _FakeBeamComponent(base_beam_component.BaseBeamComponent):
 
     def __init__(
         self,
-        type: Type[types.Artifact],  # pylint: disable=redefined-builtin
+        type: Type[types.Artifact],  # noqa: A002
         spec_kwargs: Dict[str, Any]):
       spec = _FakeComponentSpec(output=types.Channel(type=type), **spec_kwargs)
       super().__init__(spec=spec)

Diff for: tfx/types/artifact_property.py

@@ -49,7 +49,7 @@ class Property:
       PropertyType.BOOLEAN: metadata_store_pb2.BOOLEAN,
   }
 
-  def __init__(self, type):  # pylint: disable=redefined-builtin
+  def __init__(self, type):  # noqa: A002
     if type not in Property._ALLOWED_MLMD_TYPES:
       raise ValueError('Property type must be one of %s.' %
                        list(Property._ALLOWED_MLMD_TYPES.keys()))

Diff for: tfx/types/channel.py

@@ -123,7 +123,7 @@ class BaseChannel(abc.ABC, Generic[_AT]):
       set.
   """
 
-  def __init__(self, type: Type[_AT], is_optional: Optional[bool] = None):  # pylint: disable=redefined-builtin
+  def __init__(self, type: Type[_AT], is_optional: Optional[bool] = None):  # noqa: A002
     if not _is_artifact_type(type):
       raise ValueError(
           'Argument "type" of BaseChannel constructor must be a subclass of '
@@ -155,11 +155,11 @@ def as_optional(self) -> typing_extensions.Self:
     return new_channel
 
   @property
-  def type(self) -> Type[_AT]:  # pylint: disable=redefined-builtin
+  def type(self) -> Type[_AT]:  # noqa: A002
     return self._artifact_type
 
   @type.setter
-  def type(self, value: Type[_AT]):  # pylint: disable=redefined-builtin
+  def type(self, value: Type[_AT]):  # noqa: A002
     self._set_type(value)
 
   @doc_controls.do_not_generate_docs
@@ -235,7 +235,7 @@ class won't be removed in TFX 1.x due to backward compatibility guarantee
   # TODO(b/125348988): Add support for real Channel in addition to static ones.
   def __init__(
       self,
-      type: Type[Artifact],  # pylint: disable=redefined-builtin
+      type: Type[Artifact],  # noqa: A002
       additional_properties: Optional[Dict[str, Property]] = None,
       additional_custom_properties: Optional[Dict[str, Property]] = None,
       # TODO(b/161490287): deprecate static artifact.

Diff for: tfx/types/component_spec.py

@@ -340,7 +340,7 @@ class MyCustomComponentSpec(ComponentSpec):
       non-primitive types like lists) should be stored in its original form.
   """
 
-  def __init__(self, type=None, optional=False, use_proto=False):  # pylint: disable=redefined-builtin
+  def __init__(self, type=None, optional=False, use_proto=False):  # noqa: A002
     self.type = type
     self.optional = optional
     self.use_proto = use_proto
@@ -419,7 +419,7 @@ class MyCustomComponentSpec(ComponentSpec):
 
   def __init__(
       self,
-      type: Optional[Type[artifact.Artifact]] = None,  # pylint: disable=redefined-builtin
+      type: Optional[Type[artifact.Artifact]] = None,  # noqa: A002
       optional: bool = False,
       allow_empty: Optional[bool] = None,
       is_async: bool = False,

Diff for: tfx/v1/dsl/io/fileio.py

@@ -22,7 +22,7 @@
 from tfx.dsl.io.fileio import makedirs
 from tfx.dsl.io.fileio import mkdir
 from tfx.dsl.io.fileio import NotFoundError
-from tfx.dsl.io.fileio import open  # pylint: disable=redefined-builtin
+from tfx.dsl.io.fileio import open  # noqa: A002
 from tfx.dsl.io.fileio import PathType
 from tfx.dsl.io.fileio import remove
 from tfx.dsl.io.fileio import rename

Diff for: tfx/v1/dsl/placeholders/__init__.py

@@ -16,7 +16,7 @@
 
 from tfx.dsl.placeholder.placeholder import exec_property
 from tfx.dsl.placeholder.placeholder import execution_invocation
-from tfx.dsl.placeholder.placeholder import input  # pylint: disable=redefined-builtin
+from tfx.dsl.placeholder.placeholder import input  # noqa: A002
 from tfx.dsl.placeholder.placeholder import output
 
 __all__ = [