We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c4e0924 commit e35460cCopy full SHA for e35460c
source/tests/tf/test_out_bias_std.py
@@ -7,6 +7,9 @@
7
from deepmd.tf.descriptor.se_a import (
8
DescrptSeA,
9
)
10
+from deepmd.tf.env import (
11
+ tf,
12
+)
13
from deepmd.tf.fit.dipole import (
14
DipoleFittingSeA,
15
@@ -21,6 +24,16 @@
21
24
class TestOutBiasStd(unittest.TestCase):
22
25
"""Test out_bias and out_std functionality in TensorFlow backend."""
23
26
27
+ def setUp(self):
28
+ """Resets the default graph before each test."""
29
+ super().setUp()
30
+ tf.reset_default_graph()
31
+
32
+ def tearDown(self):
33
+ """Resets the default graph after each test."""
34
35
+ super().tearDown()
36
37
def test_init_out_stat_basic(self):
38
"""Test basic init_out_stat functionality."""
39
descriptor = DescrptSeA(
0 commit comments