Fix IndexError bug (#978)

Catch IndexError

Fix IndexError bug (#978)
Catch IndexError
44f54866 · Lee · xuehui · 54118c34 · 44f54866
Commit 44f54866 authored Apr 16, 2019 by Lee Committed by xuehui Apr 16, 2019
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/sdk/pynni/nni/hyperopt_tuner/hyperopt_tuner.py src/sdk/pynni/nni/hyperopt_tuner/hyperopt_tuner.py +1 -1

No files found.
--- a/src/sdk/pynni/nni/hyperopt_tuner/hyperopt_tuner.py
+++ b/src/sdk/pynni/nni/hyperopt_tuner/hyperopt_tuner.py
@@ -347,7 +347,7 @@ class HyperoptTuner(Tuner):
        for key in vals:
            try:
                parameter[key] = vals[key][0].item()
-            except KeyError:
+            except (KeyError, IndexError):
                parameter[key] = None

        # remove '_index' from json2parameter and save params-id