From 19ea13688692aaed15cb2c426dab89faeffe9eba Mon Sep 17 00:00:00 2001
From: ayushis <ayushis@netflix.com>
Date: Wed, 30 Oct 2024 09:35:17 -0700
Subject: [PATCH] remove EBS from the suggestions

---
 service_capacity_modeling/models/org/netflix/cassandra.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/service_capacity_modeling/models/org/netflix/cassandra.py b/service_capacity_modeling/models/org/netflix/cassandra.py
index 4661445..0b2aeb2 100644
--- a/service_capacity_modeling/models/org/netflix/cassandra.py
+++ b/service_capacity_modeling/models/org/netflix/cassandra.py
@@ -224,6 +224,10 @@ def _estimate_cassandra_cluster_zonal(
     if instance.cpu < 2 or instance.ram_gib < 14:
         return None
 
+    #temporarily dont suggest EBS instances
+    if instance.drive is None:
+        return None
+
     # if we're not allowed to use gp2, skip EBS only types
     if instance.drive is None and require_local_disks:
         return None