thestage 0.5.471__py3-none-any.whl → 0.6.2__py3-none-any.whl

thestage/services/clients/thestage_api/dtos/enums/container_status.py

@@ -1,17 +1,17 @@
-from enum import Enum
-
-
-class DockerContainerStatus(str, Enum):
-    FAILED: str = 'FAILED'
-    BUSY: str = 'BUSY'
-    DEAD: str = 'DEAD'
-    CREATING: str = 'CREATING'
-    CREATING_FAILED: str = 'CREATING_FAILED'
-    STARTING: str = 'STARTING'
-    RUNNING: str = 'RUNNING'
-    STOPPING: str = 'STOPPING'
-    STOPPED: str = 'STOPPED'
-    RESTARTING: str = 'RESTARTING'
-    DELETING: str = 'DELETING'
-    DELETED: str = 'DELETED'
-    UNKNOWN: str = 'UNKNOWN'
+from enum import Enum
+
+
+class DockerContainerStatus(str, Enum):
+    FAILED: str = 'FAILED'
+    BUSY: str = 'BUSY'
+    DEAD: str = 'DEAD'
+    CREATING: str = 'CREATING'
+    CREATING_FAILED: str = 'CREATING_FAILED'
+    STARTING: str = 'STARTING'
+    RUNNING: str = 'RUNNING'
+    STOPPING: str = 'STOPPING'
+    STOPPED: str = 'STOPPED'
+    RESTARTING: str = 'RESTARTING'
+    DELETING: str = 'DELETING'
+    DELETED: str = 'DELETED'
+    UNKNOWN: str = 'UNKNOWN'

thestage/services/clients/thestage_api/dtos/enums/cpu_type.py

@@ -1,8 +1,8 @@
-from enum import Enum
-
-
-class InstanceCpuType(str, Enum):
-    INTEL: str = 'INTEL'
-    AMD: str = 'AMD'
-    ARM: str = 'ARM'
-    UNKNOWN: str = 'UNKNOWN'
+from enum import Enum
+
+
+class InstanceCpuType(str, Enum):
+    INTEL: str = 'INTEL'
+    AMD: str = 'AMD'
+    ARM: str = 'ARM'
+    UNKNOWN: str = 'UNKNOWN'

thestage/services/clients/thestage_api/dtos/enums/currency_type.py

@@ -1,10 +1,10 @@
-from enum import Enum
-
-
-class CurrencyTypeEnumDto(str, Enum):
-    ETH: str = 'ETH'
-    STAGI: str = 'STAGI'
-    USD: str = 'USD'
-    EUR: str = 'EUR'
-    USDT: str = 'USDT'
-    UNKNOWN: str = 'UNKNOWN'
+from enum import Enum
+
+
+class CurrencyTypeEnumDto(str, Enum):
+    ETH: str = 'ETH'
+    STAGI: str = 'STAGI'
+    USD: str = 'USD'
+    EUR: str = 'EUR'
+    USDT: str = 'USDT'
+    UNKNOWN: str = 'UNKNOWN'

thestage/services/clients/thestage_api/dtos/enums/daemon_status.py

@@ -1,9 +1,9 @@
-from enum import Enum
-
-
-class DaemonStatus(str, Enum):
-    NEW: str = 'NEW'
-    ONLINE: str = 'ONLINE'
-    OFFLINE: str = 'OFFLINE'
-    UNKNOWN: str = 'UNKNOWN'
-    ALL: str = 'ALL'
+from enum import Enum
+
+
+class DaemonStatus(str, Enum):
+    NEW: str = 'NEW'
+    ONLINE: str = 'ONLINE'
+    OFFLINE: str = 'OFFLINE'
+    UNKNOWN: str = 'UNKNOWN'
+    ALL: str = 'ALL'

thestage/services/clients/thestage_api/dtos/enums/disk_type.py

@@ -1,7 +1,7 @@
-from enum import Enum
-
-
-class DiskTypeEnumDto(str, Enum):
-    SSD: str = 'SSD'
-    HDD: str = 'HDD'
-    UNKNOWN: str = 'UNKNOWN'
+from enum import Enum
+
+
+class DiskTypeEnumDto(str, Enum):
+    SSD: str = 'SSD'
+    HDD: str = 'HDD'
+    UNKNOWN: str = 'UNKNOWN'

thestage/services/clients/thestage_api/dtos/enums/drive_type.py

@@ -1,7 +1,7 @@
-from enum import Enum
-
-
-class DriveTypeEnumDto(str, Enum):
-    SSD: str = 'SSD'
-    HDD: str = 'HDD'
-    UNKNOWN: str = 'UNKNOWN'
+from enum import Enum
+
+
+class DriveTypeEnumDto(str, Enum):
+    SSD: str = 'SSD'
+    HDD: str = 'HDD'
+    UNKNOWN: str = 'UNKNOWN'

thestage/services/clients/thestage_api/dtos/enums/gpu_name.py

@@ -1,8 +1,8 @@
-from enum import Enum
-
-
-class InstanceGpuType(str, Enum):
-    NVIDIA: str = 'NVIDIA'
-    AMD: str = 'AMD'
-    NO_GPU: str = 'NO_GPU'
-    UNKNOWN: str = 'UNKNOWN'
+from enum import Enum
+
+
+class InstanceGpuType(str, Enum):
+    NVIDIA: str = 'NVIDIA'
+    AMD: str = 'AMD'
+    NO_GPU: str = 'NO_GPU'
+    UNKNOWN: str = 'UNKNOWN'

thestage/services/clients/thestage_api/dtos/enums/inference_model_status.py

@@ -1,10 +1,10 @@
-from enum import Enum
-
-
-class InferenceModelStatus(str, Enum):
-    SCHEDULED: str = 'SCHEDULED'
-    PROCESSING: str = 'PROCESSING'
-    PUSH_SUCCEED: str = 'PUSH_SUCCEED'
-    PUSH_FAILED: str = 'PUSH_FAILED'
-    UNKNOWN: str = 'UNKNOWN'
+from enum import Enum
+
+
+class InferenceModelStatus(str, Enum):
+    SCHEDULED: str = 'SCHEDULED'
+    PROCESSING: str = 'PROCESSING'
+    PUSH_SUCCEED: str = 'PUSH_SUCCEED'
+    PUSH_FAILED: str = 'PUSH_FAILED'
+    UNKNOWN: str = 'UNKNOWN'
     ALL: str = 'ALL'

thestage/services/clients/thestage_api/dtos/enums/inference_simulator_status.py

@@ -1,16 +1,16 @@
-from enum import Enum
-
-
-class InferenceSimulatorStatus(str, Enum):
-    SCHEDULED: str = 'SCHEDULED'
-    CREATING: str = 'CREATING'
-    CREATING_FAILED: str = 'CREATING_FAILED'
-    RUNNING: str = 'RUNNING'
-    STOPPING: str = 'STOPPING'
-    STOPPED: str = 'STOPPED'
-    DELETING: str = 'DELETING'
-    DELETED: str = 'DELETED'
-    FAILED: str = 'FAILED'
-    RESTARTING: str = 'RESTARTING'
-    UNKNOWN: str = 'UNKNOWN'
+from enum import Enum
+
+
+class InferenceSimulatorStatus(str, Enum):
+    SCHEDULED: str = 'SCHEDULED'
+    CREATING: str = 'CREATING'
+    CREATING_FAILED: str = 'CREATING_FAILED'
+    RUNNING: str = 'RUNNING'
+    STOPPING: str = 'STOPPING'
+    STOPPED: str = 'STOPPED'
+    DELETING: str = 'DELETING'
+    DELETED: str = 'DELETED'
+    FAILED: str = 'FAILED'
+    RESTARTING: str = 'RESTARTING'
+    UNKNOWN: str = 'UNKNOWN'
     ALL: str = 'ALL'

thestage/services/clients/thestage_api/dtos/enums/instance_rented_status.py

@@ -1,17 +1,17 @@
-from enum import Enum
-
-
-class InstanceRentedBusinessStatus(str, Enum):
-    IN_QUEUE: str = 'IN_QUEUE'
-    CREATING: str = 'CREATING'
-    ONLINE: str = 'ONLINE'
-    TERMINATING: str = 'TERMINATING'
-    STOPPED: str = 'STOPPED'
-    STOPPING: str = 'STOPPING'
-    STARTING: str = 'STARTING'
-    REBOOTING: str = 'REBOOTING'
-    UNREACHABLE_DAEMON: str = 'UNREACHABLE_DAEMON'
-    DELETED: str = 'DELETED'
-    RENTAL_ERROR: str = 'RENTAL_ERROR'
-    UNKNOWN: str = 'UNKNOWN'
-    ALL: str = 'ALL'
+from enum import Enum
+
+
+class InstanceRentedBusinessStatus(str, Enum):
+    IN_QUEUE: str = 'IN_QUEUE'
+    CREATING: str = 'CREATING'
+    ONLINE: str = 'ONLINE'
+    TERMINATING: str = 'TERMINATING'
+    STOPPED: str = 'STOPPED'
+    STOPPING: str = 'STOPPING'
+    STARTING: str = 'STARTING'
+    REBOOTING: str = 'REBOOTING'
+    UNREACHABLE_DAEMON: str = 'UNREACHABLE_DAEMON'
+    DELETED: str = 'DELETED'
+    RENTAL_ERROR: str = 'RENTAL_ERROR'
+    UNKNOWN: str = 'UNKNOWN'
+    ALL: str = 'ALL'

thestage/services/clients/thestage_api/dtos/enums/instance_type.py

@@ -1,7 +1,7 @@
-from enum import Enum
-
-
-class InstanceTypeEnumDto(str, Enum):
-    RENTABLE: str = 'RENTABLE'
-    SELF_HOSTED: str = 'SELF_HOSTED'
-    UNKNOWN: str = 'UNKNOWN'
+from enum import Enum
+
+
+class InstanceTypeEnumDto(str, Enum):
+    RENTABLE: str = 'RENTABLE'
+    SELF_HOSTED: str = 'SELF_HOSTED'
+    UNKNOWN: str = 'UNKNOWN'

thestage/services/clients/thestage_api/dtos/enums/location_region.py

@@ -1,11 +1,11 @@
-from enum import Enum
-
-
-class LocationRegionEnumDto(str, Enum):
-    NORTH_AMERICA: str = 'NORTH_AMERICA'
-    EUROPE: str = 'EUROPE'
-    ASIA: str = 'ASIA'
-    AFRICA: str = 'AFRICA'
-    OCEANIA: str = 'OCEANIA'
-    SOUTH_AMERICA: str = 'SOUTH_AMERICA'
-    UNKNOWN: str = 'UNKNOWN'
+from enum import Enum
+
+
+class LocationRegionEnumDto(str, Enum):
+    NORTH_AMERICA: str = 'NORTH_AMERICA'
+    EUROPE: str = 'EUROPE'
+    ASIA: str = 'ASIA'
+    AFRICA: str = 'AFRICA'
+    OCEANIA: str = 'OCEANIA'
+    SOUTH_AMERICA: str = 'SOUTH_AMERICA'
+    UNKNOWN: str = 'UNKNOWN'

thestage/services/clients/thestage_api/dtos/enums/power_status.py

@@ -1,10 +1,10 @@
-from enum import Enum
-
-
-class PowerStatusEnumDto(str, Enum):
-    ON: str = 'ON'
-    OFF: str = 'OFF'
-    POWERING_ON: str = 'POWERING_ON'
-    POWERING_OFF: str = 'POWERING_OFF'
-    REBOOTING: str = 'REBOOTING'
-    UNKNOWN: str = 'UNKNOWN'
+from enum import Enum
+
+
+class PowerStatusEnumDto(str, Enum):
+    ON: str = 'ON'
+    OFF: str = 'OFF'
+    POWERING_ON: str = 'POWERING_ON'
+    POWERING_OFF: str = 'POWERING_OFF'
+    REBOOTING: str = 'REBOOTING'
+    UNKNOWN: str = 'UNKNOWN'

thestage/services/clients/thestage_api/dtos/enums/provider_name.py

@@ -1,11 +1,11 @@
-from enum import Enum
-
-
-class ProviderNameEnumDto(str, Enum):
-    Amazon: str = 'Amazon'
-    DigitalOcean: str = 'DigitalOcean' # old
-    Exoscale: str = 'Exoscale'  # old
-    Vultr: str = 'Vultr'    # old
-    Nebius: str = 'Nebius'
-    Azure: str = 'Azure'
-    UNKNOWN: str = 'UNKNOWN'
+from enum import Enum
+
+
+class ProviderNameEnumDto(str, Enum):
+    Amazon: str = 'Amazon'
+    DigitalOcean: str = 'DigitalOcean' # old
+    Exoscale: str = 'Exoscale'  # old
+    Vultr: str = 'Vultr'    # old
+    Nebius: str = 'Nebius'
+    Azure: str = 'Azure'
+    UNKNOWN: str = 'UNKNOWN'

thestage/services/clients/thestage_api/dtos/enums/selfhosted_status.py

@@ -1,10 +1,10 @@
-from enum import Enum
-
-
-class SelfhostedBusinessStatus(str, Enum):
-    AWAITING_CONFIGURATION: str = 'AWAITING_CONFIGURATION'
-    RUNNING: str = 'RUNNING'
-    UNREACHABLE_DAEMON: str = 'UNREACHABLE_DAEMON'
-    DELETED: str = 'DELETED'
-    UNKNOWN: str = 'UNKNOWN'
-    ALL: str = 'ALL'
+from enum import Enum
+
+
+class SelfhostedBusinessStatus(str, Enum):
+    AWAITING_CONFIGURATION: str = 'AWAITING_CONFIGURATION'
+    RUNNING: str = 'RUNNING'
+    UNREACHABLE_DAEMON: str = 'UNREACHABLE_DAEMON'
+    DELETED: str = 'DELETED'
+    UNKNOWN: str = 'UNKNOWN'
+    ALL: str = 'ALL'

thestage/services/clients/thestage_api/dtos/enums/task_execution_status.py

@@ -1,12 +1,12 @@
-from enum import Enum
-
-
-class TaskExecutionStatusEnumDto(str, Enum):
-    NEW: str = 'NEW'
-    RUNNING: str = 'RUNNING'
-    FINISHED: str = 'FINISHED'
-    CANCELED: str = 'CANCELED'
-    CANCELING: str = 'CANCELING'
-    SCHEDULED: str = 'SCHEDULED'
-    FAILED: str = 'FAILED'
-    UNKNOWN: str = 'UNKNOWN'
+from enum import Enum
+
+
+class TaskExecutionStatusEnumDto(str, Enum):
+    NEW: str = 'NEW'
+    RUNNING: str = 'RUNNING'
+    FINISHED: str = 'FINISHED'
+    CANCELED: str = 'CANCELED'
+    CANCELING: str = 'CANCELING'
+    SCHEDULED: str = 'SCHEDULED'
+    FAILED: str = 'FAILED'
+    UNKNOWN: str = 'UNKNOWN'

thestage/services/clients/thestage_api/dtos/enums/task_status.py

@@ -1,12 +1,12 @@
-from enum import Enum
-from typing import List
-
-
-class TaskStatus(str, Enum):
-    NEW: str = 'NEW'
-    SCHEDULED: str = 'SCHEDULED'
-    RUNNING: str = 'RUNNING'
-    FINISHED: str = 'FINISHED'
-    CANCELING: str = 'CANCELING'
-    CANCELED: str = 'CANCELED'
-    FAILED: str = 'FAILED'
+from enum import Enum
+from typing import List
+
+
+class TaskStatus(str, Enum):
+    NEW: str = 'NEW'
+    SCHEDULED: str = 'SCHEDULED'
+    RUNNING: str = 'RUNNING'
+    FINISHED: str = 'FINISHED'
+    CANCELING: str = 'CANCELING'
+    CANCELED: str = 'CANCELED'
+    FAILED: str = 'FAILED'

thestage/services/clients/thestage_api/dtos/frontend_status.py

@@ -1,10 +1,10 @@
-from typing import Optional
-
-from pydantic import ConfigDict, BaseModel, Field
-
-
-class FrontendStatusDto(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    status_key: Optional[str] = Field(None, alias='statusKey')
-    status_translation: Optional[str] = Field(None, alias='statusTranslation')
+from typing import Optional
+
+from pydantic import ConfigDict, BaseModel, Field
+
+
+class FrontendStatusDto(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    status_key: Optional[str] = Field(None, alias='statusKey')
+    status_translation: Optional[str] = Field(None, alias='statusTranslation')

thestage/services/clients/thestage_api/dtos/inference_controller/deploy_inference_model_to_instance_request.py

@@ -1,14 +1,14 @@
-from typing import Optional, List
-
-from pydantic import Field, ConfigDict, BaseModel
-
-from thestage.services.clients.thestage_api.dtos.entity_filter_request import EntityFilterRequest
-
-
-class DeployInferenceModelToInstanceRequest(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    modelSlug: Optional[str] = Field(None, alias='modelSlug')
-    instanceRentedSlug: Optional[str] = Field(None, alias='instanceRentedSlug')
-    selfhostedInstanceSlug: Optional[str] = Field(None, alias='selfhostedInstanceSlug')
+from typing import Optional, List
+
+from pydantic import Field, ConfigDict, BaseModel
+
+from thestage.services.clients.thestage_api.dtos.entity_filter_request import EntityFilterRequest
+
+
+class DeployInferenceModelToInstanceRequest(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    modelSlug: Optional[str] = Field(None, alias='modelSlug')
+    instanceRentedSlug: Optional[str] = Field(None, alias='instanceRentedSlug')
+    selfhostedInstanceSlug: Optional[str] = Field(None, alias='selfhostedInstanceSlug')
     inferenceSimulatorSlug: Optional[str] = Field(None, alias='inferenceSimulatorSlug')

thestage/services/clients/thestage_api/dtos/inference_controller/deploy_inference_model_to_instance_response.py

@@ -1,13 +1,13 @@
-from typing import Optional
-
-from pydantic import Field, ConfigDict
-
-from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
-from thestage.services.project.dto.inference_simulator_model_dto import InferenceSimulatorModelDto
-
-
-class DeployInferenceModelToInstanceResponse(TheStageBaseResponse):
-    model_config = ConfigDict(use_enum_values=True)
-    inferenceSimulatorModel: Optional[InferenceSimulatorModelDto] = Field(None, alias='inferenceSimulatorModel')
-
-
+from typing import Optional
+
+from pydantic import Field, ConfigDict
+
+from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
+from thestage.services.project.dto.inference_simulator_model_dto import InferenceSimulatorModelDto
+
+
+class DeployInferenceModelToInstanceResponse(TheStageBaseResponse):
+    model_config = ConfigDict(use_enum_values=True)
+    inferenceSimulatorModel: Optional[InferenceSimulatorModelDto] = Field(None, alias='inferenceSimulatorModel')
+
+

thestage/services/clients/thestage_api/dtos/inference_controller/deploy_inference_model_to_sagemaker_request.py

@@ -1,12 +1,12 @@
-from typing import Optional, List
-
-from pydantic import Field, ConfigDict, BaseModel
-
-from thestage.services.clients.thestage_api.dtos.entity_filter_request import EntityFilterRequest
-
-
-class DeployInferenceModelToSagemakerRequest(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    modelSlug: Optional[str] = Field(None, alias='modelSlug')
-    arn: Optional[str] = Field(None, alias='arn')
+from typing import Optional, List
+
+from pydantic import Field, ConfigDict, BaseModel
+
+from thestage.services.clients.thestage_api.dtos.entity_filter_request import EntityFilterRequest
+
+
+class DeployInferenceModelToSagemakerRequest(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    modelSlug: Optional[str] = Field(None, alias='modelSlug')
+    arn: Optional[str] = Field(None, alias='arn')

thestage/services/clients/thestage_api/dtos/inference_controller/deploy_inference_model_to_sagemaker_response.py

@@ -1,12 +1,12 @@
-from pydantic import Field, ConfigDict
-
-from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
-
-
-class DeployInferenceModelToSagemakerResponse(TheStageBaseResponse):
-    model_config = ConfigDict(use_enum_values=True)
-    modelId: str = Field(None, alias='modelId')
-    ecrImageUrl: str = Field(None, alias='ecrImageUrl')
-    s3ArtifactsUrl: str = Field(None, alias='s3ArtifactsUrl')
-
-
+from pydantic import Field, ConfigDict
+
+from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
+
+
+class DeployInferenceModelToSagemakerResponse(TheStageBaseResponse):
+    model_config = ConfigDict(use_enum_values=True)
+    modelId: str = Field(None, alias='modelId')
+    ecrImageUrl: str = Field(None, alias='ecrImageUrl')
+    s3ArtifactsUrl: str = Field(None, alias='s3ArtifactsUrl')
+
+

thestage/services/clients/thestage_api/dtos/inference_controller/get_inference_simulator_request.py

@@ -1,11 +1,11 @@
-from typing import Optional, List
-
-from pydantic import Field, ConfigDict, BaseModel
-
-from thestage.services.clients.thestage_api.dtos.entity_filter_request import EntityFilterRequest
-
-
-class GetInferenceSimulatorRequest(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
+from typing import Optional, List
+
+from pydantic import Field, ConfigDict, BaseModel
+
+from thestage.services.clients.thestage_api.dtos.entity_filter_request import EntityFilterRequest
+
+
+class GetInferenceSimulatorRequest(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
     slug: Optional[str] = Field(None, alias='slug')

thestage/services/clients/thestage_api/dtos/inference_controller/get_inference_simulator_response.py

@@ -1,13 +1,13 @@
-from typing import Optional, List
-
-from pydantic import Field, ConfigDict, BaseModel
-
-from thestage.services.clients.thestage_api.dtos.entity_filter_request import EntityFilterRequest
-from thestage.services.project.dto.inference_simulator_dto import InferenceSimulatorDto
-
-
-class GetInferenceSimulatorResponse(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-    inferenceSimulator: Optional[InferenceSimulatorDto] = Field(None, alias='inferenceSimulator')
-
-
+from typing import Optional, List
+
+from pydantic import Field, ConfigDict, BaseModel
+
+from thestage.services.clients.thestage_api.dtos.entity_filter_request import EntityFilterRequest
+from thestage.services.project.dto.inference_simulator_dto import InferenceSimulatorDto
+
+
+class GetInferenceSimulatorResponse(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+    inferenceSimulator: Optional[InferenceSimulatorDto] = Field(None, alias='inferenceSimulator')
+
+

thestage/services/clients/thestage_api/dtos/inference_controller/inference_simulator_list_for_project_request.py

@@ -1,14 +1,14 @@
-from typing import Optional, List
-
-from pydantic import Field, ConfigDict, BaseModel
-
-from thestage.services.clients.thestage_api.dtos.entity_filter_request import EntityFilterRequest
-
-
-class InferenceSimulatorListForProjectRequest(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    entityFilterRequest: EntityFilterRequest = Field(None, alias='entityFilterRequest')
-    projectSlug: str = Field(None, alias='projectSlug')
-    statuses: Optional[List[str]] = Field(None, alias='statuses')
-
+from typing import Optional, List
+
+from pydantic import Field, ConfigDict, BaseModel
+
+from thestage.services.clients.thestage_api.dtos.entity_filter_request import EntityFilterRequest
+
+
+class InferenceSimulatorListForProjectRequest(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    entityFilterRequest: EntityFilterRequest = Field(None, alias='entityFilterRequest')
+    projectSlug: str = Field(None, alias='projectSlug')
+    statuses: Optional[List[str]] = Field(None, alias='statuses')
+

thestage/services/clients/thestage_api/dtos/inference_controller/inference_simulator_list_for_project_response.py

@@ -1,12 +1,12 @@
-
-from pydantic import Field, ConfigDict
-
-from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
-from thestage.services.clients.thestage_api.dtos.paginated_entity_list import PaginatedEntityList
-from thestage.services.project.dto.inference_simulator_dto import InferenceSimulatorDto
-
-
-class InferenceSimulatorListForProjectResponse(TheStageBaseResponse):
-    model_config = ConfigDict(use_enum_values=True)
-
-    inferenceSimulators: PaginatedEntityList[InferenceSimulatorDto] = Field(None, alias='inferenceSimulators')
+
+from pydantic import Field, ConfigDict
+
+from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
+from thestage.services.clients.thestage_api.dtos.paginated_entity_list import PaginatedEntityList
+from thestage.services.project.dto.inference_simulator_dto import InferenceSimulatorDto
+
+
+class InferenceSimulatorListForProjectResponse(TheStageBaseResponse):
+    model_config = ConfigDict(use_enum_values=True)
+
+    inferenceSimulators: PaginatedEntityList[InferenceSimulatorDto] = Field(None, alias='inferenceSimulators')