conftest.py 359 B

123456789101112
  1. import pytest
  2. @pytest.fixture()
  3. def should_do_global_cleanup_after_test() -> bool:
  4. """Disable the global cleanup fixture for tests in this directory. This
  5. provides a ~10x speedup for unit tests that don't load a model to GPU.
  6. This requires that tests in this directory clean up after themselves if they
  7. use the GPU.
  8. """
  9. return False