Merge pull request #356 from vespa-engine/tgm/increase-container-stop…

…-timeout Fix Vespa Cloud integration tests failure due to Vespa 8 tensor default formatting
vespa-engine · Jun 23, 2022 · c748bcb · c748bcb
2 parents ec19e3c + 9c03aa7
commit c748bcb
Showing 1 changed file with 4 additions and 11 deletions.
diff --git a/vespa/test_integration_vespa_cloud.py b/vespa/test_integration_vespa_cloud.py
@@ -209,13 +209,8 @@ def setUp(self) -> None:
             expected_fields.update(
                 {
                     "pretrained_bert_tiny_doc_token_ids": {
-                        "cells": [
-                            {
-                                "address": {"d0": str(x)},
-                                "value": float(tensor_field_values[x]),
-                            }
-                            for x in range(len(tensor_field_values))
-                        ]
+                        "type": f"tensor<float>(d0[{len(tensor_field_values)}])",
+                        "values": tensor_field_values,
                     }
                 }
             )
@@ -331,10 +326,8 @@ def setUp(self) -> None:
                 "dataset": d["dataset"],
                 "context_id": d["context_id"],
                 "sentence_embedding": {
-                    "cells": [
-                        {"address": {"x": str(idx)}, "value": value}
-                        for idx, value in enumerate(d["sentence_embedding"]["values"])
-                    ]
+                    "type": f"tensor<float>(x[{len(d['sentence_embedding']['values'])}])",
+                    "values": d["sentence_embedding"]["values"],
                 },
             }
             if len(d["questions"]) > 0: