keras/conftest.py
2023-06-26 16:06:26 -07:00

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