Skip to content

Commit 1801651

Browse files
committed
[keras/legacy_tf_layers/migration_utils.py] Move docstring from method to class and document seed
1 parent 5331dac commit 1801651

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

keras/legacy_tf_layers/migration_utils.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -43,14 +43,14 @@ class DeterministicRandomTestTool(object):
4343
This applies both to the stateful random operations used for creating and
4444
initializing variables, and to the stateful random operations used in
4545
computation (such as for dropout layers).
46+
47+
Args:
48+
mode: Set mode to 'constant' or 'num_random_ops'. Defaults to
49+
'constant'.
50+
seed: The random seed to use.
4651
"""
4752

4853
def __init__(self, seed: int = 42, mode="constant"):
49-
"""
50-
Args:
51-
mode: Set mode to 'constant' or 'num_random_ops'. Defaults to
52-
'constant'.
53-
"""
5454
if mode not in {"constant", "num_random_ops"}:
5555
raise ValueError(
5656
"Mode arg must be 'constant' or 'num_random_ops'. "

0 commit comments

Comments
 (0)