Skip to content

Commit 560af86

Browse files
author
Raymond Liu
committed
fix inf.py
1 parent c7dbe02 commit 560af86

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

src/sagemaker/serve/model_server/torchserve/xgboost_inference.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -91,6 +91,8 @@ def output_fn(predictions, accept_type):
9191

9292

9393
def _run_preflight_diagnostics():
94+
install_package("sagemaker")
95+
install_package("boto3", "1.17.52")
9496
_py_vs_parity_check()
9597
_pickle_file_integrity_check()
9698

@@ -129,9 +131,5 @@ def install_package(package_name, version=None):
129131
print(f"Failed to install {package_name}. Error: {e}")
130132

131133

132-
# Example usage
133-
install_package("sagemaker")
134-
install_package("boto3", "1.17.52")
135-
136134
# on import, execute
137135
_run_preflight_diagnostics()

0 commit comments

Comments
 (0)