Skip to content

Commit

Permalink
update docsting on settings
Browse files Browse the repository at this point in the history
  • Loading branch information
strickvl committed Aug 25, 2024
1 parent 9bf9a15 commit de8eb01
Showing 1 changed file with 11 additions and 0 deletions.
11 changes: 11 additions & 0 deletions src/zenml/integrations/modal/flavors/modal_step_operator_flavor.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,18 +26,29 @@
class ModalStepOperatorSettings(BaseSettings):
"""Settings for the Modal step operator.
Specifying the region and cloud provider is only available for Enterprise
and Team plan customers.
Certain combinations of settings are not available. It is suggested to err
on the side of looser settings rather than more restrictive ones to avoid
pipeline execution failures. In the case of failures, however, Modal
provides detailed error messages that can help identify what is
incompatible. See more in the Modal docs at https://modal.com/docs/guide/region-selection.
Attributes:
gpu: The type of GPU to use for the step execution.
cpu: The number of CPU cores to allocate for the step execution.
memory: The amount of memory (in megabytes) to allocate for the step
execution.
region: The region to use for the step execution.
cloud: The cloud provider to use for the step execution.
"""

gpu: Optional[str] = None
cpu: Optional[float] = None
memory: Optional[int] = None
region: Optional[str] = None
cloud: Optional[str] = None


class ModalStepOperatorConfig(
Expand Down

0 comments on commit de8eb01

Please sign in to comment.