Fix t2, t3 logic

ddps-lab · Feb 14, 2025 · f97007b · f97007b
1 parent 54c3426
commit f97007b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/collector/spot-dataset/aws/ec2/sps/sps_query_api.py b/collector/spot-dataset/aws/ec2/sps/sps_query_api.py
@@ -70,11 +70,11 @@ def query_sps(args):
             sps_dict["AZ"].append(score['AvailabilityZoneId'])
             sps_dict["SPS"].append(int(score["Score"]))
             sps_dict["TargetCapacity"].append(target_capacity)
-            if score['AvailabilityZoneId'] == 3:
+            if score['Score'] == 3:
                 sps_dict["T3"].append(target_capacity)
             else:
                 sps_dict["T3"].append(0)
-            if score['AvailabilityZoneId'] == 1:
+            if score['Score'] == 1:
                 sps_dict["T2"].append(0)
             else:
                 sps_dict["T2"].append(target_capacity)