service-capacity-modeling 0.3.60__py3-none-any.whl → 0.3.62__py3-none-any.whl

service_capacity_modeling/capacity_planner.py

@@ -586,7 +586,6 @@ class CapacityPlanner:
 
     # Calculates the minimum cpu, memory, and network requirements based on desires.
     def _per_instance_requirements(self, desires) -> Tuple[int, float]:
-
         # Applications often set fixed reservations of heap or OS memory
         per_instance_mem = (
             desires.data_shape.reserved_instance_app_mem_gib

service_capacity_modeling/hardware/__init__.py

@@ -110,8 +110,7 @@ def merge_hardware(existing: Hardware, override: Hardware) -> Hardware:
             for shape in existing_keys | override_keys:
                 if shape in existing_keys and shape in override_keys:
                     raise ValueError(
-                        f"Duplicate shape {shape}! "
-                        "Only one file should contain a shape"
+                        f"Duplicate shape {shape}! Only one file should contain a shape"
                     )
                 if shape not in existing_keys:
                     merged_field[shape] = override_field[shape]

service_capacity_modeling/interface.py

@@ -405,7 +405,7 @@ class Instance(ExcludeUnsetModel):
         self_dict = self.model_dump()
         other_dict = overrides.model_dump(exclude_unset=True)
 
-        for (k, v) in other_dict.items():
+        for k, v in other_dict.items():
             # TODO we need a deep merge on drive (recursive merge)
             if k in ("platforms",):
                 # Unique merge platforms

service_capacity_modeling/models/common.py

@@ -355,9 +355,8 @@ def compute_stateful_zone(  # pylint: disable=too-many-positional-arguments
     # or less IOs for a given data size as well as space
     # Contract for disk ios is
     # (per_node_size_gib, node_count) -> (read_ios, write_ios)
-    required_disk_ios: Callable[
-        [float, int], Tuple[float, float]
-    ] = lambda size_gib, count: (0, 0),
+    required_disk_ios: Callable[[float, int], Tuple[float, float]] = lambda size_gib,
+    count: (0, 0),
     required_disk_space: Callable[[float], float] = lambda size_gib: size_gib,
     # The maximum amount of state we can hold per node in the database
     # typically you don't want stateful systems going much higher than a

service_capacity_modeling/models/org/netflix/cassandra.py

@@ -13,7 +13,6 @@ from service_capacity_modeling.interface import AccessConsistency
 from service_capacity_modeling.interface import AccessPattern
 from service_capacity_modeling.interface import Buffer
 from service_capacity_modeling.interface import BufferComponent
-from service_capacity_modeling.interface import BufferIntent
 from service_capacity_modeling.interface import Buffers
 from service_capacity_modeling.interface import CapacityDesires
 from service_capacity_modeling.interface import CapacityPlan
@@ -603,7 +602,7 @@ class NflxCassandraCapacityModel(CapacityModel):
                 f"Required cluster size must be at least "
                 f"{CRITICAL_TIER_MIN_CLUSTER_SIZE=} when "
                 f"service tier({tier}) is a "
-                f"critical tier({CRITICAL_TIERS})."
+                f"critical tier({CRITICAL_TIERS}). "
                 f"If it is an existing cluster, horizontally "
                 f"scale the cluster to be >= "
                 f"{CRITICAL_TIER_MIN_CLUSTER_SIZE}"
@@ -629,10 +628,10 @@ class NflxCassandraCapacityModel(CapacityModel):
         require_attached_disks: bool = extra_model_arguments.get(
             "require_attached_disks", False
         )
-        required_cluster_size: Optional[
-            int
-        ] = NflxCassandraCapacityModel.get_required_cluster_size(
-            desires.service_tier, extra_model_arguments
+        required_cluster_size: Optional[int] = (
+            NflxCassandraCapacityModel.get_required_cluster_size(
+                desires.service_tier, extra_model_arguments
+            )
         )
 
         max_rps_to_disk: int = extra_model_arguments.get("max_rps_to_disk", 500)

service_capacity_modeling/models/org/netflix/crdb.py

@@ -143,7 +143,6 @@ def _estimate_cockroachdb_cluster_zonal(  # noqa=E501 pylint: disable=too-many-p
     min_vcpu_per_instance: int = 4,
     license_fee_per_core: float = 0.0,
 ) -> Optional[CapacityPlan]:
-
     if instance.cpu < min_vcpu_per_instance:
         return None
 

service_capacity_modeling/models/org/netflix/ddb.py

@@ -191,9 +191,9 @@ def _get_write_consistency_percentages(
             transactional_write_percent = 0.0
             non_transactional_write_percent = 1.0
     total_percent = transactional_write_percent + non_transactional_write_percent
-    assert (
-        total_percent == 1
-    ), "transactional_write_percent, non_transactional_write_percent should sum to 1"
+    assert total_percent == 1, (
+        "transactional_write_percent, non_transactional_write_percent should sum to 1"
+    )
     return {
         "transactional_write_percent": transactional_write_percent,
         "non_transactional_write_percent": non_transactional_write_percent,
@@ -452,9 +452,9 @@ class NflxDynamoDBCapacityModel(CapacityModel):
             "data_transfer_gib": data_transfer_plan.total_data_transfer_gib,
             "target_utilization_percentage": target_util_percentage,
         }
-        requirement_context[
-            "replicated_write_capacity_units"
-        ] = write_plan.replicated_write_capacity_units
+        requirement_context["replicated_write_capacity_units"] = (
+            write_plan.replicated_write_capacity_units
+        )
 
         dynamo_costs = {
             "dynamo.regional-writes": write_plan.total_annual_write_cost,
@@ -462,9 +462,9 @@ class NflxDynamoDBCapacityModel(CapacityModel):
             "dynamo.regional-storage": storage_plan.total_annual_data_storage_cost,
         }
 
-        dynamo_costs[
-            "dynamo.regional-transfer"
-        ] = data_transfer_plan.total_annual_data_transfer_cost
+        dynamo_costs["dynamo.regional-transfer"] = (
+            data_transfer_plan.total_annual_data_transfer_cost
+        )
 
         dynamo_costs["dynamo.data-backup"] = backup_plan.total_annual_backup_cost
 

service_capacity_modeling/models/org/netflix/elasticsearch.py

@@ -184,7 +184,6 @@ class NflxElasticsearchDataCapacityModel(CapacityModel):
         desires: CapacityDesires,
         extra_model_arguments: Dict[str, Any],
     ) -> Optional[CapacityPlan]:
-
         copies_per_region: int = _target_rf(
             desires, extra_model_arguments.get("copies_per_region", None)
         )

service_capacity_modeling/models/org/netflix/evcache.py

@@ -217,7 +217,6 @@ def _estimate_evcache_cluster_zonal(  # noqa: C901,E501 pylint: disable=too-many
     min_instance_memory_gib: int = 12,
     cross_region_replication: Replication = Replication.none,
 ) -> Optional[CapacityPlan]:
-
     # EVCache doesn't like to deploy on single CPU instances
     if instance.cpu < 2:
         return None

service_capacity_modeling/models/org/netflix/graphkv.py

@@ -30,7 +30,6 @@ class NflxGraphKVCapacityModel(CapacityModel):
         desires: CapacityDesires,
         extra_model_arguments: Dict[str, Any],
     ) -> Optional[CapacityPlan]:
-
         graphkv_app = nflx_java_app_capacity_model.capacity_plan(
             instance=instance,
             drive=drive,

service_capacity_modeling/models/org/netflix/kafka.py

@@ -110,10 +110,8 @@ def _estimate_kafka_requirement(  # pylint: disable=too-many-positional-argument
         (write_mib_per_second * MIB_IN_BYTES) * copies_per_region
     ) / MEGABIT_IN_BYTES
     bw_out = (
-        (
-            (read_mib_per_second * MIB_IN_BYTES)
-            + ((write_mib_per_second * MIB_IN_BYTES) * (copies_per_region - 1))
-        )
+        (read_mib_per_second * MIB_IN_BYTES)
+        + ((write_mib_per_second * MIB_IN_BYTES) * (copies_per_region - 1))
     ) / MEGABIT_IN_BYTES
     if (
         current_zonal_capacity
@@ -252,7 +250,6 @@ def _estimate_kafka_cluster_zonal(  # noqa: C901
     min_instance_memory_gib: int = 12,
     require_same_instance_family: bool = True,
 ) -> Optional[CapacityPlan]:
-
     # Kafka doesn't like to deploy on single CPU instances or with < 12 GiB of ram
     if instance.cpu < min_instance_cpu or instance.ram_gib < min_instance_memory_gib:
         return None
@@ -469,7 +466,6 @@ class NflxKafkaArguments(BaseModel):
 
 
 class NflxKafkaCapacityModel(CapacityModel):
-
     HA_DEFAULT_REPLICATION_FACTOR = 2
     SC_DEFAULT_REPLICATION_FACTOR = 3
 

service_capacity_modeling/models/org/netflix/stateless_java.py

@@ -91,7 +91,6 @@ def _estimate_java_app_region(  # pylint: disable=too-many-positional-arguments
     failover: bool = True,
     jvm_memory_overhead: float = 2,
 ) -> Optional[CapacityPlan]:
-
     if drive.name != "gp2":
         return None
 

service_capacity_modeling/models/org/netflix/wal.py

@@ -30,7 +30,6 @@ class NflxWALCapacityModel(CapacityModel):
         desires: CapacityDesires,
         extra_model_arguments: Dict[str, Any],
     ) -> Optional[CapacityPlan]:
-
         wal_app = nflx_java_app_capacity_model.capacity_plan(
             instance=instance,
             drive=drive,

service_capacity_modeling/models/org/netflix/zookeeper.py

@@ -35,7 +35,6 @@ def _zk_requirement(
     heap_overhead: float,
     disk_overhead: float,
 ) -> Optional[CapacityRequirement]:
-
     # We only deploy Zookeeper to fast ephemeral storage
     # Due to fsync latency to the disk.
     if instance.drive is None:
@@ -94,7 +93,6 @@ class NflxZookeeperCapacityModel(CapacityModel):
         desires: CapacityDesires,
         extra_model_arguments: Dict[str, Any],
     ) -> Optional[CapacityPlan]:
-
         # We only deploy Zookeeper to 3 zone regions at this time
         if context.zones_in_region != 3:
             return None

service_capacity_modeling/tools/auto_shape.py

@@ -178,9 +178,9 @@ def pull_family(
                     # The biggest size drive is the single tenant one
                     disk_max_size = max(disk_max_size, int(disk["SizeInGB"]))
 
-            instance_jsons_dict[
-                instance_type_json["InstanceType"].split(".")[1]
-            ] = instance_type_json
+            instance_jsons_dict[instance_type_json["InstanceType"].split(".")[1]] = (
+                instance_type_json
+            )
 
     if disk_type.name.startswith("local"):
         if (

service_capacity_modeling/tools/fetch_pricing.py

@@ -22,7 +22,6 @@ def extract_3yr_upfront_price(price_data):
             and term_attrs.get("PurchaseOption") == "All Upfront"
             and term_attrs.get("OfferingClass") == "standard"
         ):
-
             # Get upfront fee
             for dim in term["priceDimensions"].values():
                 if dim["unit"] == "Quantity":

{service_capacity_modeling-0.3.60.dist-info → service_capacity_modeling-0.3.62.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: service-capacity-modeling
-Version: 0.3.60
+Version: 0.3.62
 Summary: Contains utilities for modeling capacity for pluggable workloads
 Author: Joseph Lynch
 Author-email: josephl@netflix.com
@@ -251,16 +251,28 @@ tox -e py38 -- -k test_<your_functionality> --pdb --pdbcls=IPython.terminal.debu
 ```
 
 ### Pre-commit / Linting
-To install the pre-commit linter
-```
-pre-commit install
-```
-
 To run the linting manually:
 ```
 tox -e pre-commit
 ```
 
+### Installing Pre-commit Hooks
+
+This repository includes a custom pre-commit hook that runs all linting and formatting checks through the tox environment. To install it:
+
+```bash
+# Install the custom pre-commit hook
+tox -e install-hooks
+
+# Or manually copy the hook
+cp hooks/pre-commit .git/hooks/pre-commit
+chmod +x .git/hooks/pre-commit
+```
+
+The hook will automatically:
+- Create the tox pre-commit environment if it doesn't exist
+- Run all pre-commit checks (ruff, flake8, etc.)
+- Ensure all code quality standards are met before commits
 
 ### PyCharm IDE Setup
 Use one of the test environments for IDE development, e.g. `tox -e py310` and then

{service_capacity_modeling-0.3.60.dist-info → service_capacity_modeling-0.3.62.dist-info}/RECORD

@@ -1,8 +1,8 @@
 service_capacity_modeling/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-service_capacity_modeling/capacity_planner.py,sha256=B-C6EgZEWLKS6HEL74WxGeoLHrzNN6Cmyfee2jKsdcU,31915
-service_capacity_modeling/interface.py,sha256=145YV4Lm2YZhBAKjsdUIjeIOxJ7NxqUPwz9oxekgsPo,36995
+service_capacity_modeling/capacity_planner.py,sha256=B6e0esOAvV6qMkEeLIO9rEveTftRG_Ut_d9gYgSIM0w,31914
+service_capacity_modeling/interface.py,sha256=eJRahrVv-JMbalwd6gqVLbtvCoBhZSLyG-mm8asOBD4,36993
 service_capacity_modeling/stats.py,sha256=8HIPwVnmvbauBwXhn6vbNYO7-CzWPuymnq0eX7ZA1_w,5849
-service_capacity_modeling/hardware/__init__.py,sha256=Lc-DdX2ZJ_sFT3KlImPUUcKmcjVnavH_mMXlyCwpNiw,9001
+service_capacity_modeling/hardware/__init__.py,sha256=kzIHnIymwnf4qQYDpfIChIAxTF8b87XtnBg1TwF_J9E,8974
 service_capacity_modeling/hardware/profiles/__init__.py,sha256=7-y3JbCBkgzaAjFla2RIymREcImdZ51HTl3yn3vzoGw,1602
 service_capacity_modeling/hardware/profiles/profiles.txt,sha256=tOfSR3B0E0uAOaXd5SLI3ioq83UYZ3yhK7UHhsK4awQ,49
 service_capacity_modeling/hardware/profiles/pricing/aws/3yr-reserved_ec2.json,sha256=JNAj4yotSrEzlsMLamg_GmhqOiiTwKNcDPNs44PTLxI,52798
@@ -40,38 +40,38 @@ service_capacity_modeling/hardware/profiles/shapes/aws/manual_drives.json,sha256
 service_capacity_modeling/hardware/profiles/shapes/aws/manual_instances.json,sha256=i611n6d7hsjd7L8aSEDzfaTMS2jVs-Jc38-vl-NKfs4,18013
 service_capacity_modeling/hardware/profiles/shapes/aws/manual_services.json,sha256=h63675KKmu5IrI3BORDN8fiAqLjAyYHArErKbC7-T30,776
 service_capacity_modeling/models/__init__.py,sha256=XK7rTBW8ZXQY5L9Uy2FwjuFN_KBW3hKw7IrhG1piajs,13567
-service_capacity_modeling/models/common.py,sha256=ixmNkohvw30R2JBvdpM6xKb19na_g3GNGvQQYLlkV_A,33577
+service_capacity_modeling/models/common.py,sha256=f2M3ZpQopENexwwDeoMyH57oktenz88advR2XQoNTys,33567
 service_capacity_modeling/models/headroom_strategy.py,sha256=QIkP_K_tK2EGAjloaGfXeAPH5M0UDCN8FlAtwV9xxTA,651
 service_capacity_modeling/models/utils.py,sha256=0F__wz9KAGhPIQfvNp-FTtTANW6-sO4FsyddnuXqSJc,2161
 service_capacity_modeling/models/org/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 service_capacity_modeling/models/org/netflix/__init__.py,sha256=m7IaQbo85NEbDvfoPJREIznpzg0YHTCrKP5C1GvnOYM,2378
 service_capacity_modeling/models/org/netflix/aurora.py,sha256=Mi9zd48k64GkKIjAs3J1S2qThguNvyWIy2dUmhwrVhc,12883
-service_capacity_modeling/models/org/netflix/cassandra.py,sha256=ppvHfHIwSzljlkSDTbPBXXaX0N1suVdmI3sfEwN7Cek,34682
+service_capacity_modeling/models/org/netflix/cassandra.py,sha256=BsMx9QYSGSh4wKdJm_dR5rBHdbYlE7NMNWiYIQ0oU4w,34632
 service_capacity_modeling/models/org/netflix/counter.py,sha256=hOVRRCgCPU-A5TdLKQXc_mWTQpkKOWRNjOeECdDP7kA,9205
-service_capacity_modeling/models/org/netflix/crdb.py,sha256=2rD4Io0yT7o0NR4lNferXXOSTDe0SkT1LbSChvNgRrQ,19698
-service_capacity_modeling/models/org/netflix/ddb.py,sha256=2jxMFz31xckJvymvVlu1yWm0X4dGYlqxDo0bftU1B9M,26307
-service_capacity_modeling/models/org/netflix/elasticsearch.py,sha256=Ku7OFz-6BY0FS0lfNh9MCqtDcjufLcuPbug0CO4UPvY,23686
+service_capacity_modeling/models/org/netflix/crdb.py,sha256=AlHdGFpR1RmwQSZsiuiHLR2wTrnmtguT2MMYBDHfdiM,19697
+service_capacity_modeling/models/org/netflix/ddb.py,sha256=GDoXVIpDDY6xDB0dsiaz7RAPPj-qffTrM9N6w5-5ndg,26311
+service_capacity_modeling/models/org/netflix/elasticsearch.py,sha256=746WYY_WSeMgY-Go7wvWGYOxfFV8ryupNThvDZHPbGo,23685
 service_capacity_modeling/models/org/netflix/entity.py,sha256=M0vzwhf8UAbVxnXspAkN4GEbq3rix6yoky6W2oDG6a0,8648
-service_capacity_modeling/models/org/netflix/evcache.py,sha256=l3fzIVQ1PznD2tG00Fmwl728437MZrtHOS4WjXVa3fs,25229
-service_capacity_modeling/models/org/netflix/graphkv.py,sha256=EVRo-1OCDvvotqPgIdP2_JXMfZUsG7KZxRMmlgYc3CI,8558
+service_capacity_modeling/models/org/netflix/evcache.py,sha256=G5d8OXp21FslKt2EhYWxhxicKEggA0lQhG3Bid93elw,25228
+service_capacity_modeling/models/org/netflix/graphkv.py,sha256=iS5QDDv9_hNY6nIgdL-umB439qP7-jN-n6_Tl6d-ZSo,8557
 service_capacity_modeling/models/org/netflix/iso_date_math.py,sha256=CPGHLmbGeNqkcYcmCkLKhPZcAU-yTJ2HjvuXdnNyCYc,996
-service_capacity_modeling/models/org/netflix/kafka.py,sha256=cIvO88g7YhEOYsMAVbZ5a4YyJCvbKaTmwQjk73gg_7s,25545
+service_capacity_modeling/models/org/netflix/kafka.py,sha256=FiuBml8uWOVfPFZ37NvZW13nMVtFE4VDf_SZIFdw4sA,25515
 service_capacity_modeling/models/org/netflix/key_value.py,sha256=yL5moU0SJD4ocBU9zeGhPYE4KY7oSSq5yqfVWd_Ot2g,9336
 service_capacity_modeling/models/org/netflix/postgres.py,sha256=R3Tog-ZW1Yx6gO3AKqI_wquSm30s01QX9yWR7Jvgk9A,4055
 service_capacity_modeling/models/org/netflix/rds.py,sha256=z9egFBg4Ltqyuz_WHk-_hw-xL-EQNzl1JopJoWdNli8,10768
-service_capacity_modeling/models/org/netflix/stateless_java.py,sha256=WYIDUdy-1BgVTLcAbMW2xg-Xx6RPusITLOKNh2DuA1w,11307
+service_capacity_modeling/models/org/netflix/stateless_java.py,sha256=1la-z5tG5CGr7QM5SHBGguxzkA7BxJPAoUWhthADe8s,11306
 service_capacity_modeling/models/org/netflix/time_series.py,sha256=XDujp-m7bRQWFDXKY7nto0OQUq3v1z56HZKs3eO8o_I,8167
 service_capacity_modeling/models/org/netflix/time_series_config.py,sha256=Qrtngn6VQq13NF55h_oRsSNA8lVnOOIb-c-dsQn6TH4,7311
-service_capacity_modeling/models/org/netflix/wal.py,sha256=EdhEKTqxdMRobvqp5EHtjnNE4UpWCt1rNXKQUx5OlL8,4410
-service_capacity_modeling/models/org/netflix/zookeeper.py,sha256=buYv400R8_KCF4Y02uQ5tzcBCvk5jSGfPnMWSh4A9EI,7603
+service_capacity_modeling/models/org/netflix/wal.py,sha256=9GA7V5pZF1mKu9mM7kI9XN_i6U7Ap1oh6REzXO2ZxOM,4409
+service_capacity_modeling/models/org/netflix/zookeeper.py,sha256=BHLjnVDyx15wMGrc0QFmv9v6M95snU30WR7mhIRoa4Q,7601
 service_capacity_modeling/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-service_capacity_modeling/tools/auto_shape.py,sha256=Rk5Fjrw2susVL8It_J2KUADoMGBN6N394fcThdl62Ng,14672
-service_capacity_modeling/tools/fetch_pricing.py,sha256=SHOtFaPr61op2bnY9i_g_1-d-Nz2rV8c7Jwsye2R49s,3763
+service_capacity_modeling/tools/auto_shape.py,sha256=41pfR40BN-xJS8js4BWSoqT67JhO2_XqzmNeKDoCFBo,14674
+service_capacity_modeling/tools/fetch_pricing.py,sha256=JkgJPTE0SVj8sdGQvo0HN-Hdv3nfA2tu7C_Arad5aX8,3762
 service_capacity_modeling/tools/generate_missing.py,sha256=uvr9fQanx3bm4KTneH-x7EOQvO7cVV0i9gdQvArPCuY,2947
 service_capacity_modeling/tools/instance_families.py,sha256=e9JWSIdljSmHI8Nb2MI5Ld9JqQ7WdOtPtV7g3oR7ZiU,7764
-service_capacity_modeling-0.3.60.dist-info/licenses/LICENSE,sha256=nl_Lt5v9VvJ-5lWJDT4ddKAG-VZ-2IaLmbzpgYDz2hU,11343
-service_capacity_modeling-0.3.60.dist-info/METADATA,sha256=tMXT9YuVyoSpvawrjPjzLWQTwIJUXHrr9_XmL2btp7Y,9733
-service_capacity_modeling-0.3.60.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-service_capacity_modeling-0.3.60.dist-info/entry_points.txt,sha256=ZsjzpG5SomWpT1zCE19n1uSXKH2gTI_yc33sdl0vmJg,146
-service_capacity_modeling-0.3.60.dist-info/top_level.txt,sha256=H8XjTCLgR3enHq5t3bIbxt9SeUkUT8HT_SDv2dgIT_A,26
-service_capacity_modeling-0.3.60.dist-info/RECORD,,
+service_capacity_modeling-0.3.62.dist-info/licenses/LICENSE,sha256=nl_Lt5v9VvJ-5lWJDT4ddKAG-VZ-2IaLmbzpgYDz2hU,11343
+service_capacity_modeling-0.3.62.dist-info/METADATA,sha256=3PzrFsAG7No9MPYyMQ-MVY28MHZLysYPUX4MFcCmzsU,10214
+service_capacity_modeling-0.3.62.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+service_capacity_modeling-0.3.62.dist-info/entry_points.txt,sha256=ZsjzpG5SomWpT1zCE19n1uSXKH2gTI_yc33sdl0vmJg,146
+service_capacity_modeling-0.3.62.dist-info/top_level.txt,sha256=H8XjTCLgR3enHq5t3bIbxt9SeUkUT8HT_SDv2dgIT_A,26
+service_capacity_modeling-0.3.62.dist-info/RECORD,,