Changed inference.py to compile policy.sample actions

[WilliamYue37]

What this does

Changed inference.py to compile policy.sample actions. It was compiling the forward method before.

How it was tested

Ran GPU pytests and regression tests.

How to checkout & try? (for the reviewer)

Provide a simple way for the reviewer to try out your changes.

python src/opentau/scripts/inference.py --config_path=/path/to/config

Checklist

• I have added Google-style docstrings to important functions and ensured function parameters are typed.
• My PR includes policy-related changes.
  • If the above is checked: I have run the GPU pytests (pytest -m "gpu") and regression tests.

Note: Before submitting this PR, please read the contributor guideline.