errors when running the last af3score step

[alexie2001xcc-dev]

I0605 08:59:06.271519 131487940564800 pipeline.py:165] processing p53_p53_4_8, random_seed=10
I0605 08:59:06.280114 131487940564800 pipeline.py:258] Calculating bucket size for input with 93 tokens.
I0605 08:59:06.280187 131487940564800 pipeline.py:264] Got bucket size 93 for input with 93 tokens, resulting in 0 padded tokens.
Featurising p53_p53_4_8 with rng_seed 10 took 0.92 seconds.
First inference - triggering JAX compilation...
Error processing fold_input in p53_p53_4_8.json: 'JITFunction' object has no attribute '_get_config'. Skipping...
Task p53_p53_4_8 completed in 2.07s
Clearing position data cache...
Clearing JAX compilation cache...
Cache cleanup completed
Batch processing completed, total time: 127.62 seconds
Average per file: 1.60 seconds
All processing completed successfully.
[AF3scoreStep] Step 5/5: extracting metrics
Traceback (most recent call last):
File "/home/xcc/software/PPIFlow/pipeline.py", line 1867, in
pipeline.run(stage=args.stage)
File "/home/xcc/software/PPIFlow/pipeline.py", line 1803, in run
self.run_stage1()
File "/home/xcc/software/PPIFlow/pipeline.py", line 1333, in run_stage1
).run()
^^^^^
File "/home/xcc/software/PPIFlow/steps/af3score_step.py", line 1098, in run
self._extract_metrics()
File "/home/xcc/software/PPIFlow/steps/af3score_step.py", line 1018, in _extract_metrics
self._write_metrics_csv(rows)
File "/home/xcc/software/PPIFlow/steps/af3score_step.py", line 1007, in _write_metrics_csv
raise RuntimeError("No metrics were extracted.")
RuntimeError: No metrics were extracted.

[MacroHongZ]

First inference - triggering JAX compilation...
Error processing fold_input in p53_p53_4_8.json: 'JITFunction' object has no attribute '_get_config'. 

This is a JAX error. Please test the af3score step separately.