Skip to content

Commit 68f7aac

Browse files
committed
Fix error message and test
1 parent 256955e commit 68f7aac

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

src/model.c

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -653,7 +653,7 @@ int RedisAI_Parse_ModelRun_RedisCommand(RedisModuleCtx *ctx,
653653
RedisModule_ReplyWithError(
654654
ctx,
655655
"Number of names given as OUTPUTS during MODELSET and keys given as "
656-
"INPUTS here do not match");
656+
"OUTPUTS here do not match");
657657
}
658658
return -1;
659659
}

test/tests_tensorflow.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -454,6 +454,8 @@ def test_run_tf_model_errors(env):
454454
env.assertEqual("Insufficient arguments, missing model BLOB",exception.__str__())
455455

456456
try:
457+
con.execute_command('AI.TENSORSET', 'a{1}', 'FLOAT', '2', 'VALUES', '2', '2')
458+
con.execute_command('AI.TENSORSET', 'b{1}', 'FLOAT', '2', 'VALUES', '2', '2')
457459
con.execute_command('AI.MODELRUN', 'm{1}', 'INPUTS', 'a{1}', 'b{1}')
458460
except Exception as e:
459461
exception = e

0 commit comments

Comments
 (0)