experiments/STT
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Improve EnableExternalScorer error handling in Python and JS bindings

Browse Source
This commit is contained in:
Reuben Morais 2020-05-03 15:39:31 +02:00
parent 5deb8a2f7b
commit 48971413e5
2 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
native_client/javascript/index.js
View File

@ -35,7 +35,7 @@ function Model(aModelPath) {
const status = rets[0];
const impl = rets[1];
if (status !== 0) {
throw "CreateModel failed "+binding.ErrorCodeToErrorMessage(status)+" 0x" + status.toString(16);
throw "CreateModel failed with '"+binding.ErrorCodeToErrorMessage(status)+"' (0x" + status.toString(16) + ")";
}
this._impl = impl;
@ -76,10 +76,13 @@ Model.prototype.sampleRate = function() {
*
* @param {string} aScorerPath The path to the external scorer file.
*
* @return {number} Zero on success, non-zero on failure (invalid arguments).
* @throws on error
*/
Model.prototype.enableExternalScorer = function(aScorerPath) {
return binding.EnableExternalScorer(this._impl, aScorerPath);
const status = binding.EnableExternalScorer(this._impl, aScorerPath);
if (status !== 0) {
throw "EnableExternalScorer failed with '"+binding.ErrorCodeToErrorMessage(status)+"' (0x" + status.toString(16) + ")";
}
}
/**
@ -139,7 +142,7 @@ Model.prototype.createStream = function() {
const status = rets[0];
const ctx = rets[1];
if (status !== 0) {
throw "CreateStream failed "+binding.ErrorCodeToErrorMessage(status)+" 0x" + status.toString(16);
throw "CreateStream failed with '"+binding.ErrorCodeToErrorMessage(status)+"' (0x" + status.toString(16) + ")";
}
return new Stream(ctx);
}

7
native_client/python/__init__.py
View File

@ -81,10 +81,11 @@ class Model(object):
:param scorer_path: The path to the external scorer file.
:type scorer_path: str
:return: Zero on success, non-zero on failure.
:type: int
:throws: RuntimeError on error
"""
return deepspeech.impl.EnableExternalScorer(self._impl, scorer_path)
status = deepspeech.impl.EnableExternalScorer(self._impl, scorer_path)
if status != 0:
raise RuntimeError("EnableExternalScorer failed with '{}' (0x{:X})".format(deepspeech.impl.ErrorCodeToErrorMessage(status),status))
def disableExternalScorer(self):
"""