Update packages/aws-cdk-lib/aws-eks/lib/cluster.ts

Co-authored-by: Luca Pizzini <[email protected]>
aws · Nov 13, 2023 · 09c205c · 09c205c
1 parent 2acd9f3
commit 09c205c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/packages/aws-cdk-lib/aws-eks/lib/cluster.ts b/packages/aws-cdk-lib/aws-eks/lib/cluster.ts
@@ -1800,7 +1800,11 @@ export class Cluster extends ClusterBase {
    * @param options options for creating a new nodegroup
    */
   public addNodegroupCapacity(id: string, options?: NodegroupOptions): Nodegroup {
-    if (options?.instanceTypes?.some(i => nodeTypeForInstanceType(i) === NodeType.INFERENTIA)) {
+    const hasInferentiaInstanceType = [
+      options?.instanceType,
+      ...options?.instanceTypes ?? [],
+    ].some(i => i && nodeTypeForInstanceType(i) === NodeType.INFERENTIA);
+    if (hasInferentiaInstanceType) {
       this.addNeuronDevicePlugin();
     }
     return new Nodegroup(this, `Nodegroup${id}`, {