wehub-resource-sync
9 lines
255 B
Python
9 lines
255 B
Python
"""Run thunder tests as part of LitGPT CI"""
|
|
|
|
from litgpt.constants import _THUNDER_AVAILABLE
|
|
|
|
if _THUNDER_AVAILABLE:
|
|
from thunder.tests.test_networks import * # noqa: F403
|
|
else:
|
|
print("Skipping test_thunder_networks.py (thunder not available)")
|