Fix Ruff rule B028

Author: smokestacklightnin

tfx/orchestration/airflow/airflow_dag_runner.py

@@ -64,7 +64,9 @@ def __init__(self,
     if config and not isinstance(config, AirflowPipelineConfig):
       warnings.warn(
           'Pass config as a dict type is going to deprecated in 0.1.16. '
-          'Use AirflowPipelineConfig type instead.', PendingDeprecationWarning)
+          'Use AirflowPipelineConfig type instead.',
+          PendingDeprecationWarning,
+          stacklevel=2)
       config = AirflowPipelineConfig(airflow_dag_config=config)
     super().__init__(config)
 
@@ -123,7 +125,7 @@ def _replace_runtime_params(self, comp):
     for k, prop in comp.exec_properties.copy().items():
       if isinstance(prop, RuntimeParameter):
         # Airflow only supports string parameters.
-        if prop.ptype != str:
+        if prop.ptype is not str:
           raise RuntimeError(
               f'RuntimeParameter in Airflow does not support {prop.ptype}. The'
               'only ptype supported is string.')

tfx/orchestration/pipeline.py

@@ -414,7 +414,8 @@ def _persist_dsl_context_registry(self):
           'This is probably due to reusing component from another pipeline '
           'or interleaved pipeline definitions. Make sure each component '
           'belong to exactly one pipeline, and pipeline definitions are '
-          'separated.')
+          'separated.',
+          stacklevel=2)
 
   @property
   def inputs(self) -> Dict[str, Any]:
@@ -507,5 +508,6 @@ def enumerate_implicit_dependencies(
           warnings.warn(
               f'Node {component.id} depends on the output of node'
               f' {upstream_node_id}, but {upstream_node_id} is not included in'
-              ' the components of pipeline. Did you forget to add it?'
+              ' the components of pipeline. Did you forget to add it?',
+              stacklevel=2
           )

tfx/utils/deprecation_utils.py

@@ -185,4 +185,4 @@ class TfxDeprecationWarning(DeprecationWarning):  # pylint: disable=g-bad-except
 
 def warn_deprecated(msg):
   """Convenient method to warn TfxDeprecationWarning."""
-  warnings.warn(msg, TfxDeprecationWarning)
+  warnings.warn(msg, TfxDeprecationWarning, stacklevel=2)