Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Expose get_all_registered_operators and get_operator_arguments in the… #15364

Merged
merged 3 commits into from
Jul 4, 2019
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
CR
  • Loading branch information
larroy committed Jul 4, 2019
commit 4f6e1ef92b4103d344e5c9ded51d6e54653248ea
2 changes: 0 additions & 2 deletions tests/python/unittest/test_operator.py
Original file line number Diff line number Diff line change
Expand Up @@ -8667,13 +8667,11 @@ def test_get_operator_arguments():
operator_arguments = get_operator_arguments('Activation')
ok_(isinstance(operator_arguments, OperatorArguments))
larroy marked this conversation as resolved.
Show resolved Hide resolved
ok_(operator_arguments.names == ['data', 'act_type'])
print(operator_arguments.types)
ok_(operator_arguments.types
== ['NDArray-or-Symbol', "{'relu', 'sigmoid', 'softrelu', 'softsign', 'tanh'}, required"])
ok_(operator_arguments.narg == 2)



if __name__ == '__main__':
import nose
nose.runmodule()