Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_tensorflow
Commits
f56363ef
Unverified
Commit
f56363ef
authored
Mar 14, 2018
by
Karmel Allison
Committed by
GitHub
Mar 14, 2018
Browse files
Removing dependency on tf.test (#3599)
* Removing dependency on tf.test * New lines
parent
5d386d5b
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
6 deletions
+6
-6
official/utils/logging/hooks_helper_test.py
official/utils/logging/hooks_helper_test.py
+6
-6
No files found.
official/utils/logging/hooks_helper_test.py
View file @
f56363ef
...
...
@@ -19,14 +19,14 @@ from __future__ import absolute_import
from
__future__
import
division
from
__future__
import
print_function
import
unittest
import
tensorflow
as
tf
from
official.utils.logging
import
hooks_helper
tf
.
logging
.
set_verbosity
(
tf
.
logging
.
ERROR
)
class
BaseTest
(
tf
.
test
.
TestCase
):
class
BaseTest
(
unit
test
.
TestCase
):
def
test_raise_in_non_list_names
(
self
):
with
self
.
assertRaises
(
ValueError
):
...
...
@@ -48,15 +48,15 @@ class BaseTest(tf.test.TestCase):
self
.
assertEqual
(
returned_hook
[
0
].
__class__
.
__name__
.
lower
(),
expected_hook_name
)
def
test_get_train_hooks_
L
ogging
T
ensor
H
ook
(
self
):
def
test_get_train_hooks_
l
ogging
_t
ensor
_h
ook
(
self
):
test_hook_name
=
'LoggingTensorHook'
self
.
validate_train_hook_name
(
test_hook_name
,
'loggingtensorhook'
)
def
test_get_train_hooks_
P
rofiler
H
ook
(
self
):
def
test_get_train_hooks_
p
rofiler
_h
ook
(
self
):
test_hook_name
=
'ProfilerHook'
self
.
validate_train_hook_name
(
test_hook_name
,
'profilerhook'
)
def
test_get_train_hooks_
E
xamples
PerS
econd
H
ook
(
self
):
def
test_get_train_hooks_
e
xamples
_per_s
econd
_h
ook
(
self
):
test_hook_name
=
'ExamplesPerSecondHook'
self
.
validate_train_hook_name
(
test_hook_name
,
'examplespersecondhook'
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment