[TaskFlow] Fix pir for taskflow

[DrownFish19]

Before submitting

• Lint code. If there are lint issues, please format the code first.

# Install and register `pre-commit` in the project folder
pip install pre-commit && pre-commit install

# Process previous code files separately
pre-commit run --file XXXX.py

• Add test cases into tests folder. If there are codecov issues, please add tests cases first.

PR types

Bug fixes

PR changes

Others

Description

Fix taskflow when PIR=ON.

[paddle-bot]

Thanks for your contribution!

[ZHUI]

LGTM

[codecov]

Codecov Report

Attention: Patch coverage is 20.00000% with 4 lines in your changes missing coverage. Please review.

Project coverage is 52.51%. Comparing base (3967f76) to head (8fc77d1).
Report is 1 commits behind head on develop.

Files with missing lines	Patch %	Lines
paddlenlp/taskflow/task.py	20.00%	4 Missing ⚠️

❌ Your patch check has failed because the patch coverage (20.00%) is below the target coverage (80.00%). You can increase the patch coverage or adjust the target coverage.

Additional details and impacted files

@@             Coverage Diff             @@
##           develop    #9822      +/-   ##
===========================================
+ Coverage    52.43%   52.51%   +0.07%     
===========================================
  Files          731      730       -1     
  Lines       116411   115797     -614     
===========================================
- Hits         61037    60805     -232     
+ Misses       55374    54992     -382     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.