Skip to content
16 changes: 7 additions & 9 deletions doc/inference.md
Original file line number Diff line number Diff line change
Expand Up @@ -201,7 +201,7 @@ jumpstart_endpoint = HPJumpStartEndpoint.get(name="js-endpoint-name", namespace=
print(jumpstart_endpoint)

# Get custom endpoint details
custom_endpoint = HPEndpoint.get(endpoint_name="endpoint-custom")
custom_endpoint = HPEndpoint.get(name="endpoint-custom")
print(custom_endpoint)

```
Expand Down Expand Up @@ -231,11 +231,11 @@ from sagemaker.hyperpod.inference.hp_jumpstart_endpoint import HPJumpStartEndpoi
from sagemaker.hyperpod.inference.hp_endpoint import HPEndpoint

data = '{"inputs":"What is the capital of USA?"}'
jumpstart_endpoint = HPJumpStartEndpoint.get(endpoint_name="endpoint-jumpstart")
jumpstart_endpoint = HPJumpStartEndpoint.get(name="endpoint-jumpstart")
response = jumpstart_endpoint.invoke(body=data).body.read()
print(response)

custom_endpoint = HPEndpoint.get(endpoint_name="endpoint-custom")
custom_endpoint = HPEndpoint.get(name="endpoint-custom")
response = custom_endpoint.invoke(body=data).body.read()
print(response)
```
Expand Down Expand Up @@ -317,12 +317,10 @@ from sagemaker.hyperpod.inference.hp_jumpstart_endpoint import HPJumpStartEndpoi
from sagemaker.hyperpod.inference.hp_endpoint import HPEndpoint

# Invoke JumpStart endpoint
jumpstart_endpoint = HPJumpStartEndpoint.get(endpoint_name="endpoint-jumpstart")
print(jumpstart_endpoint.get_operator_logs(since_hours=0.1))
print(HPJumpStartEndpoint.get_operator_logs(since_hours=0.1))

# Invoke custom endpoint
custom_endpoint = HPEndpoint.get(endpoint_name="endpoint-custom")
print(custom_endpoint.get_operator_logs(since_hours=0.1))
print(HPEndpoint.get_operator_logs(since_hours=0.1))
```
````
`````
Expand All @@ -346,11 +344,11 @@ from sagemaker.hyperpod.inference.hp_jumpstart_endpoint import HPJumpStartEndpoi
from sagemaker.hyperpod.inference.hp_endpoint import HPEndpoint

# Delete JumpStart endpoint
jumpstart_endpoint = HPJumpStartEndpoint.get(endpoint_name="endpoint-jumpstart")
jumpstart_endpoint = HPJumpStartEndpoint.get(name="endpoint-jumpstart")
jumpstart_endpoint.delete()

# Delete custom endpoint
custom_endpoint = HPEndpoint.get(endpoint_name="endpoint-custom")
custom_endpoint = HPEndpoint.get(name="endpoint-custom")
custom_endpoint.delete()
```
````
Expand Down
16 changes: 12 additions & 4 deletions doc/training.md
Original file line number Diff line number Diff line change
Expand Up @@ -129,7 +129,7 @@ hyp describe hyp-pytorch-job --job-name <job-name>
from sagemaker.hyperpod.training import HyperPodPytorchJob

# Get an existing job
job = HyperPodPytorchJob.get(name="my-pytorch-job", namespace="my-namespace")
job = HyperPodPytorchJob.get(name="my-pytorch-job")

print(job)
```
Expand All @@ -147,7 +147,11 @@ hyp list-pods hyp-pytorch-job --job-name <job-name>

````{tab-item} SDK
```python
print(pytorch_job.list_pods())
from sagemaker.hyperpod.training import HyperPodPytorchJob

# List Pods for an existing job
job = HyperPodPytorchJob.get(name="my-pytorch-job")
print(job.list_pods())
```
````
`````
Expand All @@ -163,7 +167,11 @@ hyp get-logs hyp-pytorch-job --pod-name test-pytorch-job-cli-pod-0 --job-name te

````{tab-item} SDK
```python
print(pytorch_job.get_logs_from_pod("pod-name"))
from sagemaker.hyperpod.training import HyperPodPytorchJob

# Get pod logs for a job
job = HyperPodPytorchJob.get(name="my-pytorch-job")
print(job.get_logs_from_pod("pod-name"))
```
````
`````
Expand All @@ -181,7 +189,7 @@ hyp delete hyp-pytorch-job --job-name <job-name>
from sagemaker.hyperpod.training import HyperPodPytorchJob

# Get an existing job
job = HyperPodPytorchJob.get(name="my-pytorch-job", namespace="my-namespace")
job = HyperPodPytorchJob.get(name="my-pytorch-job")

# Delete the job
job.delete()
Expand Down