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

Co-authored-by: Luca Pizzini <lpizzini7@gmail.com>
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}`, {