"tools/git@developer.sourcefind.cn:OpenDAS/nni.git" did not exist on "99f7d79c7e772f7c58415b459db95fa9cd09b273"
Commit 9e40371c authored by suiguoxin's avatar suiguoxin
Browse files

add doc and test

parent a4d62e95
...@@ -8,6 +8,8 @@ Trial ...@@ -8,6 +8,8 @@ Trial
.. autofunction:: nni.get_current_parameter .. autofunction:: nni.get_current_parameter
.. autofunction:: nni.report_intermediate_result .. autofunction:: nni.report_intermediate_result
.. autofunction:: nni.report_final_result .. autofunction:: nni.report_final_result
.. autofunction:: nni.get_experiment_id
.. autofunction:: nni.get_trial_id
.. autofunction:: nni.get_sequence_id .. autofunction:: nni.get_sequence_id
......
...@@ -32,10 +32,13 @@ _last_metric = None ...@@ -32,10 +32,13 @@ _last_metric = None
def get_next_parameter(): def get_next_parameter():
return _params return _params
def get_sequence_id(): def get_experiment_id():
return 0 return 'fakeidex'
def get_trial_id(): def get_trial_id():
return 'fakeidtr'
def get_sequence_id():
return 0 return 0
def send_metric(string): def send_metric(string):
......
...@@ -38,6 +38,12 @@ class TrialTestCase(TestCase): ...@@ -38,6 +38,12 @@ class TrialTestCase(TestCase):
nni.get_next_parameter() nni.get_next_parameter()
self.assertEqual(nni.get_current_parameter('x'), 123) self.assertEqual(nni.get_current_parameter('x'), 123)
def test_get_experiment_id(self):
self.assertEqual(nni.get_experiment_id(), 'fakeidex')
def test_get_trial_id(self):
self.assertEqual(nni.get_trial_id(), 'fakeidtr')
def test_get_sequence_id(self): def test_get_sequence_id(self):
self.assertEqual(nni.get_sequence_id(), 0) self.assertEqual(nni.get_sequence_id(), 0)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment