@@ -79,7 +79,8 @@ <h1 class="title">Module <code>codeflare_sdk.cluster.config</code></h1>
79
79
template: str = f"{dir}/templates/new-template.yaml"
80
80
instascale: bool = False
81
81
envs: dict = field(default_factory=dict)
82
- image: str = "ghcr.io/foundation-model-stack/base:ray2.1.0-py38-gpu-pytorch1.12.0cu116-20221213-193103"</ code > </ pre >
82
+ image: str = "ghcr.io/foundation-model-stack/base:ray2.1.0-py38-gpu-pytorch1.12.0cu116-20221213-193103"
83
+ pull_secret: dict = field(default_factory=dict)</ code > </ pre >
83
84
</ details >
84
85
</ section >
85
86
< section >
@@ -93,7 +94,7 @@ <h2 class="section-title" id="header-classes">Classes</h2>
93
94
< dl >
94
95
< dt id ="codeflare_sdk.cluster.config.ClusterConfiguration "> < code class ="flex name class ">
95
96
< span > class < span class ="ident "> ClusterConfiguration</ span > </ span >
96
- < span > (</ span > < span > name: str, namespace: str = None, head_info: list = <factory>, machine_types: list = <factory>, min_cpus: int = 1, max_cpus: int = 1, min_worker: int = 1, max_worker: int = 1, min_memory: int = 2, max_memory: int = 2, gpu: int = 0, template: str = '/Users/aasthana/go/src/github.com/ codeflare-sdk/src/codeflare_sdk/templates/new-template.yaml', instascale: bool = False, envs: dict = <factory>, image: str = 'ghcr.io/foundation-model-stack/base:ray2.1.0-py38-gpu-pytorch1.12.0cu116-20221213-193103')</ span >
97
+ < span > (</ span > < span > name: str, namespace: str = None, head_info: list = <factory>, machine_types: list = <factory>, min_cpus: int = 1, max_cpus: int = 1, min_worker: int = 1, max_worker: int = 1, min_memory: int = 2, max_memory: int = 2, gpu: int = 0, template: str = '/home/eoingallinagh/Documents/RedHat/ codeflare-sdk/src/codeflare_sdk/templates/new-template.yaml', instascale: bool = False, envs: dict = <factory>, image: str = 'ghcr.io/foundation-model-stack/base:ray2.1.0-py38-gpu-pytorch1.12.0cu116-20221213-193103', pull_secret: dict = <factory> )</ span >
97
98
</ code > </ dt >
98
99
< dd >
99
100
< div class ="desc "> < p > This dataclass is used to specify resource requirements and other details, and
@@ -102,7 +103,8 @@ <h2 class="section-title" id="header-classes">Classes</h2>
102
103
< summary >
103
104
< span > Expand source code</ span >
104
105
</ summary >
105
- < pre > < code class ="python "> class ClusterConfiguration:
106
+ < pre > < code class ="python "> @dataclass
107
+ class ClusterConfiguration:
106
108
"""
107
109
This dataclass is used to specify resource requirements and other details, and
108
110
is passed in as an argument when creating a Cluster object.
@@ -122,7 +124,8 @@ <h2 class="section-title" id="header-classes">Classes</h2>
122
124
template: str = f"{dir}/templates/new-template.yaml"
123
125
instascale: bool = False
124
126
envs: dict = field(default_factory=dict)
125
- image: str = "ghcr.io/foundation-model-stack/base:ray2.1.0-py38-gpu-pytorch1.12.0cu116-20221213-193103"</ code > </ pre >
127
+ image: str = "ghcr.io/foundation-model-stack/base:ray2.1.0-py38-gpu-pytorch1.12.0cu116-20221213-193103"
128
+ pull_secret: dict = field(default_factory=dict)</ code > </ pre >
126
129
</ details >
127
130
< h3 > Class variables</ h3 >
128
131
< dl >
@@ -182,6 +185,10 @@ <h3>Class variables</h3>
182
185
< dd >
183
186
< div class ="desc "> </ div >
184
187
</ dd >
188
+ < dt id ="codeflare_sdk.cluster.config.ClusterConfiguration.pull_secret "> < code class ="name "> var < span class ="ident "> pull_secret</ span > : dict</ code > </ dt >
189
+ < dd >
190
+ < div class ="desc "> </ div >
191
+ </ dd >
185
192
< dt id ="codeflare_sdk.cluster.config.ClusterConfiguration.template "> < code class ="name "> var < span class ="ident "> template</ span > : str</ code > </ dt >
186
193
< dd >
187
194
< div class ="desc "> </ div >
@@ -221,6 +228,7 @@ <h4><code><a title="codeflare_sdk.cluster.config.ClusterConfiguration" href="#co
221
228
< li > < code > < a title ="codeflare_sdk.cluster.config.ClusterConfiguration.min_worker " href ="#codeflare_sdk.cluster.config.ClusterConfiguration.min_worker "> min_worker</ a > </ code > </ li >
222
229
< li > < code > < a title ="codeflare_sdk.cluster.config.ClusterConfiguration.name " href ="#codeflare_sdk.cluster.config.ClusterConfiguration.name "> name</ a > </ code > </ li >
223
230
< li > < code > < a title ="codeflare_sdk.cluster.config.ClusterConfiguration.namespace " href ="#codeflare_sdk.cluster.config.ClusterConfiguration.namespace "> namespace</ a > </ code > </ li >
231
+ < li > < code > < a title ="codeflare_sdk.cluster.config.ClusterConfiguration.pull_secret " href ="#codeflare_sdk.cluster.config.ClusterConfiguration.pull_secret "> pull_secret</ a > </ code > </ li >
224
232
< li > < code > < a title ="codeflare_sdk.cluster.config.ClusterConfiguration.template " href ="#codeflare_sdk.cluster.config.ClusterConfiguration.template "> template</ a > </ code > </ li >
225
233
</ ul >
226
234
</ li >
0 commit comments