Merge pull request #7 from nadavb/master

lookup support for big numbers
jermp · Feb 25, 2019 · 1e01025 · 1e01025
2 parents 423b1bf + c520521
commit 1e01025
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/python/tongrammodule.cpp b/python/tongrammodule.cpp
@@ -37,7 +37,15 @@ static PyObject * tongram_lookup(PyObject *self, PyObject *args){
     stl_string_adaptor adaptor;
 	uint64_t value1 = model.lookup(ngrams_space_separated, adaptor);
 
-    return Py_BuildValue("i", value1);
+//	cout << "lookup value: " << value1 << std::endl;
+
+	if (value1 == global::not_found) {
+	    return Py_BuildValue("K", 0); // unsigned long long
+	} else {
+	    return Py_BuildValue("K", value1); // unsigned long long
+	}
+
+
 }
 
 static PyMethodDef TongramMethods[] = {