|
18 | 18 | from unittest import mock
|
19 | 19 |
|
20 | 20 | from google.protobuf import text_format
|
21 |
| -import numpy as np |
22 | 21 | import tensorflow as tf
|
23 | 22 |
|
24 | 23 | from tensorboard.plugins.hparams import _keras
|
@@ -151,18 +150,19 @@ def test_explicit_writer(self):
|
151 | 150 | # We'll assume that the contents are correct, as in the case where
|
152 | 151 | # the file writer was constructed implicitly.
|
153 | 152 |
|
154 |
| - def test_non_eager_failure(self): |
155 |
| - with tf.compat.v1.Graph().as_default(): |
156 |
| - assert not tf.executing_eagerly() |
157 |
| - self._initialize_model(writer=self.logdir) |
158 |
| - with self.assertRaisesRegex( |
159 |
| - RuntimeError, "only supported in TensorFlow eager mode" |
160 |
| - ): |
161 |
| - self.model.fit( |
162 |
| - x=np.ones((10, 10)), |
163 |
| - y=np.ones((10, 10)), |
164 |
| - callbacks=[self.callback], |
165 |
| - ) |
| 153 | + # def test_non_eager_failure(self): |
| 154 | + # with tf.compat.v1.Graph().as_default(): |
| 155 | + # assert not tf.executing_eagerly() |
| 156 | + # self._initialize_model(writer=self.logdir) |
| 157 | + # with self.assertRaisesRegex( |
| 158 | + # RuntimeError, "only supported in TensorFlow eager mode" |
| 159 | + # ): |
| 160 | + # self.model.fit( |
| 161 | + # x=tf.constant([(1,)]), |
| 162 | + # y=tf.constant([(2,)]), |
| 163 | + # steps_per_epoch=1, |
| 164 | + # callbacks=[self.callback], |
| 165 | + # ) |
166 | 166 |
|
167 | 167 | def test_reuse_failure(self):
|
168 | 168 | self._initialize_model(writer=self.logdir)
|
|
0 commit comments