Skip to content

Commit 073136f

Browse files
Add conftest
1 parent 6d43a98 commit 073136f

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

tensorflow_model_analysis/conftest.py

+8
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
import pytest
2+
from tensorflow.compat.v1 import disable_v2_behavior, enable_v2_behavior
3+
4+
@pytest.fixture(scope="class")
5+
def v2_behavior():
6+
enable_v2_behavior()
7+
yield
8+
disable_v2_behavior()

0 commit comments

Comments
 (0)