diff --git a/doc/inference.md b/doc/inference.md
index f15c2bd5..2b5ba665 100644
--- a/doc/inference.md
+++ b/doc/inference.md
@@ -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)
 
 ```
@@ -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)
 ```
@@ -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))
 ```
 ````
 `````
@@ -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()
 ```
 ````
diff --git a/doc/training.md b/doc/training.md
index 181f826a..7d49ae57 100644
--- a/doc/training.md
+++ b/doc/training.md
@@ -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)
 ```
@@ -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())
 ```
 ````
 `````
@@ -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"))
 ```
 ````
 `````
@@ -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()