Skip to content

Commit 6840eb4

Browse files
Fix applications_test setup and teardown
1 parent cf9f492 commit 6840eb4

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

tf_keras/applications/applications_test.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -150,13 +150,11 @@
150150

151151

152152
class ApplicationsTest(tf.test.TestCase, parameterized.TestCase):
153-
@classmethod
154-
def setUpClass(cls):
155-
cls.original_image_data_format = backend.image_data_format()
153+
def setUp(self):
154+
self.original_image_data_format = backend.image_data_format()
156155

157-
@classmethod
158-
def tearDownClass(cls):
159-
backend.set_image_data_format(cls.original_image_data_format)
156+
def tearDown(self):
157+
backend.set_image_data_format(self.original_image_data_format)
160158

161159
@classmethod
162160
def assertShapeEqual(cls, shape1, shape2):

0 commit comments

Comments
 (0)