Skip to content

Commit

Permalink
update model file changes
Browse files Browse the repository at this point in the history
  • Loading branch information
ramrutha497 committed Apr 24, 2024
1 parent 4b0d3d2 commit 0a196c1
Showing 1 changed file with 20 additions and 10 deletions.
30 changes: 20 additions & 10 deletions spotinst_sdk2/models/ocean/gcp/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,26 +22,26 @@ def __init__(
self.max_scale_down_percentage = max_scale_down_percentage


class HeadRoom:
class Headroom:
"""
# Arguments
cpu_per_unit: int
gpu_per_unit: int
memory_per_unit: int
num_of_unit: int
num_of_units: int
"""

def __init__(
self,
cpu_per_unit: int = none,
gpu_per_unit: int = none,
memory_per_unit: int = none,
num_of_unit: int = none
num_of_units: int = none
):
self.cpu_per_unit = cpu_per_unit
self.gpu_per_unit = gpu_per_unit
self.memory_per_unit = memory_per_unit
self.num_of_unit = num_of_unit
self.num_of_units = num_of_units


class ResourceLimits:
Expand All @@ -67,7 +67,7 @@ class AutoScaler:
cooldown: int
down: Down
enable_automatic_and_manual_headroom: bool
head_room: HeadRoom
headroom: Headroom
is_auto_config: bool
is_enabled: bool
resource_limits: ResourceLimits
Expand All @@ -79,7 +79,7 @@ def __init__(
cooldown: int = none,
down: Down = none,
enable_automatic_and_manual_headroom: bool = none,
head_room: HeadRoom = none,
headroom: Headroom = none,
is_auto_config: bool = none,
is_enabled: bool = none,
resource_limits: ResourceLimits = none
Expand All @@ -88,7 +88,7 @@ def __init__(
self.cooldown = cooldown
self.down = down
self.enable_automatic_and_manual_headroom = enable_automatic_and_manual_headroom
self.head_room = head_room
self.headroom = headroom
self.is_auto_config = is_auto_config
self.is_enabled = is_enabled
self.resource_limits = resource_limits
Expand Down Expand Up @@ -695,7 +695,7 @@ class AutoScale:
def __init__(
self,
auto_headroom_percentage: int = none,
headrooms: List[HeadRoom] = none,
headrooms: List[Headroom] = none,
down: Down = none
):
self.auto_headroom_percentage = auto_headroom_percentage
Expand All @@ -718,17 +718,27 @@ def __init__(
self.min_instance_count = min_instance_count


class Config:
"""
Arguments
headrooms: List[Headroom]
"""
def __init__(self,
headrooms: List[Headroom] = none):
self.headrooms = headrooms


class VNGTasks:
"""
# Arguments
config: List[HeadRoom]
config: Config
cron_expression: str
is_enabled: bool
task_type: str
"""
def __init__(
self,
config: List[HeadRoom] = none,
config: Config = none,
cron_expression: str = none,
is_enabled: bool = none,
task_type: str = none
Expand Down

0 comments on commit 0a196c1

Please sign in to comment.