8 lines
256 B
Python
8 lines
256 B
Python
|
try:
|
||
|
|
# When using torch and tensorflow, torch needs to be imported first,
|
||
|
|
# otherwise it will segfault upon import. This should force the torch
|
||
|
|
# import to happen first for all tests.
|
||
|
|
import torch # noqa: F401
|
||
|
|
except ImportError:
|
||
|
|
pass
|