[BEAM-14353] Explore versions of pytorch to test in Tox

• Type: Sub-task
• Status: Open
• Priority: P2
• Resolution: Unresolved
• Component/s: sdk-py-core
• Labels: run-inference
• Assignee: Andy Ye
• Reporter: Andy Ye
• Created: 21/Apr/22 21:42
• Updated: 04/Jun/22 23:31

Description

We test different minor versions for pandas because our special usage of pandas in the DataFrame API leads to breakages even between minor versions. In the case of pyarrow, every release is a major version, which is meant to communicate that the API can change (https://arrow.apache.org/docs/format/Versioning.html).

We need to investigate if pytorch commonly make changes that will break us between minor versions. 

How will we keep this up to date as new version of pytorch come out?