Show PyTorch version in pytest report header

kakaobrain · May 29, 2019 · 2ebac75 · 2ebac75
1 parent 634174d
commit 2ebac75
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -5,3 +5,7 @@
 @pytest.fixture(autouse=True)
 def manual_seed_zero():
     torch.manual_seed(0)
+
+
+def pytest_report_header():
+    return 'torch: %s' % torch.__version__