Change signature checks to PyTorchError

lutzroeder · Aug 13, 2018 · c9b879b · c9b879b
1 parent cb39959
commit c9b879b
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/pytorch-model.js b/src/pytorch-model.js
@@ -27,15 +27,18 @@ class PyTorchModelFactory {
                 signature.length != magic_number.length ||
                 !signature.every((value, index) => value == magic_number[index])) 
             {
-                throw new pickle.Error('Invalid signature.');
+                callback(new PyTorchError('Invalid signature.', null));
+                return;
             }
             var protocol_version = unpickler.load();
             if (protocol_version != 1001) {
-                throw new pickle.Error("Unsupported protocol version '" + protocol_version + "'.");
+                callback(new PyTorchError("Unsupported protocol version '" + protocol_version + "'.", null));
+                return;
             }
             var sysInfo = unpickler.load();
             if (!sysInfo.little_endian) {
-                throw new pickle.Error("Unsupported system.");
+                callback(new PyTorchError('Unsupported system information.'));
+                return;
             }
 
             var functionTable = {};