[SPARK-41777] Add Integration Tests

• Type: Sub-task
• Status: Resolved
• Priority: Major
• Resolution: Fixed
• Affects Version/s: 3.4.0
• Fix Version/s: 3.4.0
• Component/s: ML, PySpark
• Assignee: Rithwik Ediga Lakhamsani
• Reporter: Rithwik Ediga Lakhamsani
• Created: 29/Dec/22 18:42
• Updated: 21/Jan/23 07:22
• Resolved: 21/Jan/23 07:22

Description

This requires us to add PyTorch as a testing dependency.

Issue Links

• links to: [Github] Pull Request #39637 (rithwik-db)
• links to: [Github] Pull Request #39637 (rithwik-db)