Fix Ruff rule A002

Author: smokestacklightnin

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 {}

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)

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
 

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,

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)
 

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

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)
 

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:

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)

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()))