We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 6d43a98 commit 073136fCopy full SHA for 073136f
tensorflow_model_analysis/conftest.py
@@ -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