Fix trajectory sampling to handle active dims

[khurram-ghani]

Related issue(s)/PRs: None

Summary

Fixes issue where trajectory sampling was ignoring active-dims in the kernel

Fully backwards compatible: yes

PR checklist

• [X] The quality checks are all passing
• [X] The bug case / new feature is covered by tests
• [ ] Any new features are well-documented (in docstrings or notebooks)