Skip to content

Commit c498848

Browse files
author
Mohamed Zeidan
committed
hyp_cli fixes
1 parent 43493ac commit c498848

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

src/sagemaker/hyperpod/cli/hyp_cli.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -180,49 +180,49 @@ class CLICommand(click.Group):
180180
cli = LazyGroup()
181181

182182
# Create subgroups, lightweight and don't trigger imports
183-
@cli.group()
183+
@cli.group(cls=CLICommand)
184184
def create():
185185
"""Create endpoints or pytorch jobs."""
186186
pass
187187

188188

189-
@cli.group(name='list')
190-
def list_cmd():
189+
@cli.group(cls=CLICommand)
190+
def list():
191191
"""List endpoints or pytorch jobs."""
192192
pass
193193

194194

195-
@cli.group()
195+
@cli.group(cls=CLICommand)
196196
def describe():
197197
"""Describe endpoints or pytorch jobs."""
198198
pass
199199

200200

201-
@cli.group()
201+
@cli.group(cls=CLICommand)
202202
def delete():
203203
"""Delete endpoints or pytorch jobs."""
204204
pass
205205

206206

207-
@cli.group()
207+
@cli.group(cls=CLICommand)
208208
def list_pods():
209209
"""List pods for endpoints or pytorch jobs."""
210210
pass
211211

212212

213-
@cli.group()
213+
@cli.group(cls=CLICommand)
214214
def get_logs():
215215
"""Get pod logs for endpoints or pytorch jobs."""
216216
pass
217217

218218

219-
@cli.group()
219+
@cli.group(cls=CLICommand)
220220
def invoke():
221221
"""Invoke model endpoints."""
222222
pass
223223

224224

225-
@cli.group()
225+
@cli.group(cls=CLICommand)
226226
def get_operator_logs():
227227
"""Get operator logs for endpoints."""
228228
pass

0 commit comments

Comments
 (0)