Added gpu types

CHANGELOG.md

@@ -2,6 +2,10 @@
 All notable changes to this project will be documented in this file.
 This project adheres to [Semantic Versioning](http://semver.org/).
 
+## [3.1.0] - 2024-05-17
+### Added
+- Added support for `GpuType` in Ocean AKS `Filters` model.
+
 ## [3.1.0] - 2024-04-24
 ### Added
 - Added support for Ocean GKE APIs.

docs/models/ocean/azure.md

@@ -364,6 +364,42 @@ An enumeration.
 <h3 id="spotinst_sdk2.models.ocean.azure.VmType.storage_optimized">storage_optimized</h3>
 
 
+<h2 id="spotinst_sdk2.models.ocean.azure.GpuType">GpuType</h2>
+
+```python
+GpuType(cls, value, names=None, *, module, qualname, type, start)
+```
+An enumeration.
+<h3 id="spotinst_sdk2.models.ocean.azure.GpuType.amd_radeon_instinct_mi25">amd_radeon_instinct_mi25</h3>
+
+
+<h3 id="spotinst_sdk2.models.ocean.azure.GpuType.nvidia_a10">nvidia_a10</h3>
+
+
+<h3 id="spotinst_sdk2.models.ocean.azure.GpuType.nvidia_tesla_a100">nvidia_tesla_a100</h3>
+
+
+<h3 id="spotinst_sdk2.models.ocean.azure.GpuType.nvidia_tesla_h100">nvidia_tesla_h100</h3>
+
+
+<h3 id="spotinst_sdk2.models.ocean.azure.GpuType.nvidia_tesla_k80">nvidia_tesla_k80</h3>
+
+
+<h3 id="spotinst_sdk2.models.ocean.azure.GpuType.nvidia_tesla_m60">nvidia_tesla_m60</h3>
+
+
+<h3 id="spotinst_sdk2.models.ocean.azure.GpuType.nvidia_tesla_p100">nvidia_tesla_p100</h3>
+
+
+<h3 id="spotinst_sdk2.models.ocean.azure.GpuType.nvidia_tesla_p40">nvidia_tesla_p40</h3>
+
+
+<h3 id="spotinst_sdk2.models.ocean.azure.GpuType.nvidia_tesla_t4">nvidia_tesla_t4</h3>
+
+
+<h3 id="spotinst_sdk2.models.ocean.azure.GpuType.nvidia_tesla_v100">nvidia_tesla_v100</h3>
+
+
 <h2 id="spotinst_sdk2.models.ocean.azure.Filters">Filters</h2>
 
 ```python
@@ -385,7 +421,9 @@ Filters(
     min_n_i_cs: int = 'd3043820717d74d9a17694c176d39733',
     series: typing.List[str] = 'd3043820717d74d9a17694c176d39733',
     vm_types:
-    typing.List[spotinst_sdk2.models.ocean.azure.VmType] = 'd3043820717d74d9a17694c176d39733'
+    typing.List[spotinst_sdk2.models.ocean.azure.VmType] = 'd3043820717d74d9a17694c176d39733',
+    gpu_types:
+    typing.List[spotinst_sdk2.models.ocean.azure.GpuType] = 'd3043820717d74d9a17694c176d39733'
 )
 ```
 
@@ -405,6 +443,7 @@ __Arguments__
 - __min_n_i_cs__: int
 - __series__: List[str]
 - __vm_types__: List[VmType]
+- __gpu_types__: List[GpuType]
 
 <h2 id="spotinst_sdk2.models.ocean.azure.VmSizes">VmSizes</h2>
 

spotinst_sdk2/models/ocean/azure/__init__.py

@@ -335,6 +335,19 @@ class VmType(Enum):
     gpu = "GPU"
 
 
+class GpuType(Enum):
+    nvidia_tesla_v100 = "nvidia-tesla-v100" 
+    amd_radeon_instinct_mi25 = "amd-radeon-instinct-mi25"
+    nvidia_a10 = "nvidia-a10" 
+    nvidia_tesla_a100 = "nvidia-tesla-a100" 
+    nvidia_tesla_k80 = "nvidia-tesla-k80" 
+    nvidia_tesla_m60 = "nvidia-tesla-m60" 
+    nvidia_tesla_p100 = "nvidia-tesla-p100" 
+    nvidia_tesla_p40 = "nvidia-tesla-p40" 
+    nvidia_tesla_t4 = "nvidia-tesla-t4" 
+    nvidia_tesla_h100 = "nvidia-tesla-h100"   
+
+
 class Filters:
     """
     # Arguments
@@ -352,6 +365,7 @@ class Filters:
     min_n_i_cs: int
     series: List[str]
     vm_types: List[VmType]
+    gpu_types: List[GpuType]
     """
 
     def __init__(
@@ -369,7 +383,8 @@ def __init__(
             min_data: int = none,
             min_n_i_cs: int = none,
             series: List[str] = none,
-            vm_types: List[VmType] = none):
+            vm_types: List[VmType] = none,
+            gpu_types: List[GpuType] = none):
         self.accelerated_networking = accelerated_networking
         self.disk_performance = disk_performance
         self.architectures = architectures
@@ -384,6 +399,7 @@ def __init__(
         self.min_n_i_cs = min_n_i_cs
         self.series = series
         self.vm_types = vm_types
+        self.gpu_types = gpu_types
 
 
 class VmSizes:

spotinst_sdk2/version.py

@@ -1 +1 @@
-__version__ = '3.1.0'
+__version__ = '3.2.0'