diff --git a/rastervision_aws_sagemaker/rastervision/aws_sagemaker/__init__.py b/rastervision_aws_sagemaker/rastervision/aws_sagemaker/__init__.py index ed29df887..da73a7766 100644 --- a/rastervision_aws_sagemaker/rastervision/aws_sagemaker/__init__.py +++ b/rastervision_aws_sagemaker/rastervision/aws_sagemaker/__init__.py @@ -12,6 +12,7 @@ def register_plugin(registry): 'cpu_inst_type', 'gpu_image', 'gpu_inst_type', + 'use_spot_instances', ])