Merge pull request #1021 from vespa-engine/andreer/make-gpu-example-work

make gpu example work

vespa/package.py

@@ -134,7 +134,7 @@ def as_lines(self) -> List[str]:
                 if isinstance(field[1], str):
                     tmp_string += f"{field[1]}"
                 else:
-                    tmp_string += f'{", ".join(field[1])}'
+                    tmp_string += f"{', '.join(field[1])}"
                 final_list.append(tmp_string)
 
         final_list.append("}")
@@ -936,10 +936,10 @@ def __repr__(self) -> str:
 
 class SecondPhaseRanking(object):
     def __init__(
-            self,
-            expression: str,
-            rerank_count: int = 100,
-            rank_score_drop_limit: Optional[float] = None,
+        self,
+        expression: str,
+        rerank_count: int = 100,
+        rank_score_drop_limit: Optional[float] = None,
     ) -> None:
         r"""
         Create a Vespa second phase ranking configuration.
@@ -985,10 +985,11 @@ def __repr__(self) -> str:
 
 class GlobalPhaseRanking(object):
     def __init__(
-            self,
-            expression: str,
-            rerank_count: int = 100,
-            rank_score_drop_limit: Optional[float] = None) -> None:
+        self,
+        expression: str,
+        rerank_count: int = 100,
+        rank_score_drop_limit: Optional[float] = None,
+    ) -> None:
         r"""
         Create a Vespa global phase ranking configuration.
 
@@ -1011,7 +1012,6 @@ def __init__(
         self.rerank_count = rerank_count
         self.rank_score_drop_limit = rank_score_drop_limit
 
-
     def __eq__(self, other: object) -> bool:
         if not isinstance(other, self.__class__):
             return NotImplemented
@@ -2043,8 +2043,9 @@ def __init__(
         ...    nodes=Nodes(
         ...        count="2",
         ...        parameters=[
-        ...            Parameter("resources", {"vcpu": "4.0", "memory": "16Gb", "disk": "125Gb"}),
-        ...            Parameter("gpu", {"count": "1", "memory": "16Gb"}),
+        ...            Parameter("resources", {"vcpu": "4.0", "memory": "16Gb", "disk": "125Gb"},
+        ...                children=[Parameter("gpu", {"count": "1", "memory": "16Gb"})]
+        ...            ),
         ...            Parameter("node", {"hostalias": "node1", "distribution-key": "0"}),
         ...        ]
         ...    )