Skip to content

Commit cc236e3

Browse files
author
Bryannah Hernandez
committed
import and unused fix
1 parent acc3cbc commit cc236e3

File tree

2 files changed

+1
-5
lines changed

2 files changed

+1
-5
lines changed

src/sagemaker/pipeline.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,8 +15,8 @@
1515

1616
from typing import Optional, Dict, List, Union
1717

18-
import sagemaker
1918
import logging
19+
import sagemaker
2020
from sagemaker import ModelMetrics, Model
2121
from sagemaker.config import (
2222
ENDPOINT_CONFIG_KMS_KEY_ID_PATH,

src/sagemaker/serve/mode/in_process_mode.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -63,11 +63,7 @@ def prepare(self):
6363

6464
def create_server(
6565
self,
66-
image: str,
67-
secret_key: str,
6866
predictor: PredictorBase,
69-
env_vars: Dict[str, str] = None,
70-
model_path: str = None,
7167
):
7268
"""Creating the server and checking ping health."""
7369
logger.info("Waiting for model server %s to start up...", self.model_server)

0 commit comments

Comments
 (0)