nod-ai / SHARK-TestSuite

Temporary home of a test suite we are evaluating
Apache License 2.0
2 stars 29 forks source link

Remove unused imports from e2eshark/pytorch/**/model.py files. #231

Closed ScottTodd closed 4 months ago

ScottTodd commented 4 months ago

Progress on https://github.com/nod-ai/SHARK-TestSuite/issues/49

According to my IDE and some local running, these imports were unused. The import torch_mlir lines are particularly problematic, as torch-mlir on its own is not distributed for downstream projects to use (there are nightly Linux packages for some developer use, but downstream projects like IREE take a source dependency on torch-mlir and ship their own packages).