Fix unit tests, format code

ucbrise · Apr 14, 2017 · 4546912 · 4546912
1 parent 6c29846
commit 4546912
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/examples/tutorial/cifar_utils.py b/examples/tutorial/cifar_utils.py
@@ -120,7 +120,7 @@ def parse_pred(p):
 
 def cifar_prediction(host, app, uid, x):
     url = "http://%s:1337/%s/predict" % (host, app)
-    req_json = json.dumps({'uid': 80, 'input': list(x)})
+    req_json = json.dumps({'uid': uid, 'input': list(x)})
     headers = {'Content-type': 'application/json'}
     start = datetime.now()
     r = requests.post(url, headers=headers, data=req_json)

diff --git a/management/clipper_manager.py b/management/clipper_manager.py
@@ -269,7 +269,7 @@ def register_application(self,
         url = "http://%s:1338/admin/add_app" % self.host
         req_json = json.dumps({
             "name": name,
-            "candidate_models": model,
+            "candidate_model_names": [model],
             "input_type": input_type,
             "default_output": str(default_output),
             "latency_slo_micros": slo_micros

diff --git a/src/libclipper/test/redis_test.cpp b/src/libclipper/test/redis_test.cpp
@@ -192,7 +192,7 @@ TEST_F(RedisTest, AddApplication) {
   // entries in it (see add_application() in redis.cpp for details on what the
   // fields are).
   EXPECT_EQ(result.size(), static_cast<size_t>(5));
-  EXPECT_EQ(str_to_models(result["candidate_models"]), model_names);
+  EXPECT_EQ(str_to_model_names(result["candidate_model_names"]), model_names);
   EXPECT_EQ(parse_input_type(result["input_type"]), input_type);
   EXPECT_EQ(result["policy"], policy);
   EXPECT_EQ(result["default_output"], default_output);

diff --git a/src/management/src/management_frontend.hpp b/src/management/src/management_frontend.hpp
@@ -250,8 +250,8 @@ class RequestHandler {
         clipper::redis::get_application(redis_connection_, app_name);
     if (existing_app_data.empty()) {
       if (clipper::redis::add_application(
-          redis_connection_, app_name, candidate_model_names, input_type,
-          selection_policy, default_output, latency_slo_micros)) {
+              redis_connection_, app_name, candidate_model_names, input_type,
+              selection_policy, default_output, latency_slo_micros)) {
         return "Success!";
       } else {
         std::stringstream ss;