wip

DvirDukhan · DvirDukhan · commit 78d67a1c073c · 2020-12-21T21:11:44.000+02:00
diff --git a/src/model.c b/src/model.c
@@ -40,7 +40,6 @@ int RAI_GetModelFromKeyspace(RedisModuleCtx *ctx, RedisModuleString *keyName, Re
     return REDISMODULE_OK;
 }
 
-
 RAI_Model *RAI_ModelCreate(RAI_Backend backend, const char *devicestr, const char *tag,
                            RAI_ModelOpts opts, size_t ninputs, const char **inputs, size_t noutputs,
                            const char **outputs, const char *modeldef, size_t modellen,
diff --git a/src/redis_ai_types/model_type.c b/src/redis_ai_types/model_type.c
diff --git a/src/redis_ai_types/model_type.h b/src/redis_ai_types/model_type.h
@@ -1,5 +1,5 @@
 #pragma once
 
-#include "../serialization_include.h"
+#include "redismodule.h"
 
 int ModelType_Register(RedisModuleCtx *ctx);
diff --git a/src/redis_ai_types/script_type.c b/src/redis_ai_types/script_type.c
@@ -1,8 +1,9 @@
 #include "script_type.h"
-#include "../AOF/rai_aof_rewrite.h"
-#include "../RDB/encoder/rai_rdb_encode.h"
-#include "../RDB/decoder/rai_rdb_decoder.h"
-#include "../RDB/decoder/decode_previous.h"
+#include "../script.h"
+#include "../serialization/AOF/rai_aof_rewrite.h"
+#include "../serialization/RDB/encoder/rai_rdb_encode.h"
+#include "../serialization/RDB/decoder/rai_rdb_decoder.h"
+#include "../serialization/RDB/decoder/decode_previous.h"
 
 extern RedisModuleType *RedisAI_ScriptType;
 
diff --git a/src/redis_ai_types/script_type.h b/src/redis_ai_types/script_type.h
@@ -1,5 +1,5 @@
 #pragma once
 
-#include "../serialization_include.h"
+#include "redismodule.h"
 
 int ScriptType_Register(RedisModuleCtx *ctx);
diff --git a/src/redis_ai_types/tensor_type.c b/src/redis_ai_types/tensor_type.c
@@ -1,8 +1,9 @@
 #include "tensor_type.h"
-#include "../AOF/rai_aof_rewrite.h"
-#include "../RDB/encoder/rai_rdb_encode.h"
-#include "../RDB/decoder/rai_rdb_decoder.h"
-#include "../RDB/decoder/decode_previous.h"
+#include "../tensor.h"
+#include "../serialization/AOF/rai_aof_rewrite.h"
+#include "../serialization/RDB/encoder/rai_rdb_encode.h"
+#include "../serialization/RDB/decoder/rai_rdb_decoder.h"
+#include "../serialization/RDB/decoder/decode_previous.h"
 
 extern RedisModuleType *RedisAI_TensorType;
 
diff --git a/src/redis_ai_types/tensor_type.h b/src/redis_ai_types/tensor_type.h
@@ -1,5 +1,5 @@
 #pragma once
 
-#include "../serialization_include.h"
+#include "serialization/serialization_include.h"
 
 int TensorType_Register(RedisModuleCtx *ctx);
diff --git a/src/redisai.c b/src/redisai.c
@@ -30,9 +30,9 @@
 #include "util/queue.h"
 #include "version.h"
 
-#include "serialization/redis_ai_types/model_type.h"
-#include "serialization/redis_ai_types/script_type.h"
-#include "serialization/redis_ai_types/tensor_type.h"
+#include "redis_ai_types/model_type.h"
+#include "redis_ai_types/script_type.h"
+#include "redis_ai_types/tensor_type.h"
 
 #define REDISAI_H_INCLUDE
 #include "redisai.h"
diff --git a/src/script.c b/src/script.c
@@ -17,7 +17,6 @@
 #include "version.h"
 #include <pthread.h>
 
-
 RAI_Script *RAI_ScriptCreate(const char *devicestr, const char *tag, const char *scriptdef,
                              RAI_Error *err) {
     if (!RAI_backends.torch.script_create) {

Original file line number	Diff line number	Diff line change
`@@ -40,7 +40,6 @@ int RAI_GetModelFromKeyspace(RedisModuleCtx ctx, RedisModuleString keyName, Re`
`40`	`40`	`return REDISMODULE_OK;`
`41`	`41`	`}`
`42`	`42`
`43`		`-`
`44`	`43`	`RAI_Model RAI_ModelCreate(RAI_Backend backend, const char devicestr, const char *tag,`
`45`	`44`	`RAI_ModelOpts opts, size_t ninputs, const char **inputs, size_t noutputs,`
`46`	`45`	`const char *outputs, const char modeldef, size_t modellen,`