Fix tensor set with int values. Solves #68

Author: lantiga

src/redisai.c

@@ -206,7 +206,7 @@ int RedisAI_TensorSet_RedisCommand(RedisModuleCtx *ctx, RedisModuleString **argv
           RAI_TensorFree(t);
           return RedisModule_ReplyWithError(ctx, "ERR invalid value");
         }
-        int ret = RAI_TensorSetValueFromDouble(t, i, val);
+        int ret = RAI_TensorSetValueFromLongLong(t, i, val);
         if (ret == -1) {
           RAI_TensorFree(t);
           return RedisModule_ReplyWithError(ctx, "ERR cannot specify values for this datatype");

test/basic_tests.py

@@ -52,6 +52,10 @@ def test_set_tensor(env):
     tensor = con.execute_command('AI.TENSORGET', 'x', 'VALUES')
     values = tensor[-1]
     env.assertEqual(values, [b'2', b'3'])
+    con.execute_command('AI.TENSORSET', 'x', 'INT32', 2, 'VALUES', 2, 3)
+    tensor = con.execute_command('AI.TENSORGET', 'x', 'VALUES')
+    values = tensor[-1]
+    env.assertEqual(values, [b'2', b'3'])
 
 def test_set_tensor_multiproc(env):
     run_test_multiproc(env, 10,