Fix bug in mark TI success api (#16524)
Fix bug in mark TI success api (#16524)

Mistakenly checking for the wrong args in TI success API. Introduced in PR https://github.com/apache/airflow/pull/16233. 
Acehaidrey/incubator-airflow,danielvdende/incubator-airflow,apache/airflow,dhuang/incubator-airflow,cfei18/incubator-airflow,Acehaidrey/incubator-airflow,danielvdende/incubator-airflow,bolkedebruin/airflow,apache/incubator-airflow,lyft/incubator-airflow,mistercrunch/airflow,lyft/incubator-airflow,mistercrunch/airflow,lyft/incubator-airflow,danielvdende/incubator-airflow,lyft/incubator-airflow,dhuang/incubator-airflow,apache/incubator-airflow,Acehaidrey/incubator-airflow,danielvdende/incubator-airflow,nathanielvarona/airflow,apache/incubator-airflow,Acehaidrey/incubator-airflow,apache/airflow,mistercrunch/airflow,nathanielvarona/airflow,cfei18/incubator-airflow,dhuang/incubator-airflow,cfei18/incubator-airflow,bolkedebruin/airflow,bolkedebruin/airflow,cfei18/incubator-airflow,apache/airflow,Acehaidrey/incubator-airflow,bolkedebruin/airflow,apache/incubator-airflow,apache/airflow,cfei18/incubator-airflow,apache/airflow,danielvdende/incubator-airflow,apache/airflow,mistercrunch/airflow,nathanielvarona/airflow,nathanielvarona/airflow,nathanielvarona/airflow,cfei18/incubator-airflow,danielvdende/incubator-airflow,nathanielvarona/airflow,Acehaidrey/incubator-airflow,bolkedebruin/airflow,dhuang/incubator-airflow