AttributeError in sagemaker.local.image

[tyrion]

On line 316 of sagemaker.local.image there is a reference to sagemaker.rl.estimator but the module is not imported. Therefore we get an error:

AttributeError: module 'sagemaker' has no attribute 'rl'

The fix is also not trivial, because if you try adding the needed import at the top of the file you incur in an import loop.

To trigger this code I created an estimator, passing to it an output_path starting with "file://" and then called its fit method.

[ChoiByungWook]

Hello @tyrion,

Can you please provide a minimal repro for us to try?

The image or dataset doesn't have to be provided.

Thanks!

[tyrion]

Hello @ChoiByungWook,
here you go:

from sagemaker import get_execution_role
from sagemaker.estimator import Estimator
from sagemaker.local import LocalSession

role = get_execution_role()

session = LocalSession()
session.config = {'local': {'local_code': True,}}

est = Estimator(
    image_name='nope',
    output_path="file:///anything",
    train_instance_type='local',
    train_instance_count=1,
    role=role,
    sagemaker_session=session,
)

est.fit()

[ChoiByungWook]

@tyrion,

Thank you so much.

I will begin by reproducing this issue.

For now, I will reference the pull request that added in the code path that caused this error: #524

[ChoiByungWook]

@tyrion,

Thank you so much for bringing this to our attention. We made our fix here: #560

Please let us know if this solves your issue.

[tyrion]

The issue seems fixed to me

Thank you @ChoiByungWook