aws-samples / sagemaker-run-notebook

Tools to run Jupyter notebooks as jobs in Amazon SageMaker - ad hoc, on a schedule, or in response to events
Apache License 2.0
142 stars 70 forks source link

Updating the S3 Paths for the Jupyter Plugin #27

Open marc-rudkowski-tm opened 3 years ago

marc-rudkowski-tm commented 3 years ago

IHAC who is using the Jupyter Plugin in Sagemaker Studio. It is a great plugin, but in the documentation there is a gap for us that I wasn't able to easily figure out in the source code:

Is there a way to change the input/output buckets using the Jupyter Plugin? It seems to use a default Sagemaker bucket. Instead I'd like to scope it to a specific bucket and prefix. Let me know if this is possible.

Cheers!

tomfaulhaber commented 3 years ago

This isn't currently possible from the plug-in, no. It shouldn't be too hard to modify the code to do it, though.

marc-rudkowski-tm commented 3 years ago

This isn't currently possible from the plug-in, no. It shouldn't be too hard to modify the code to do it, though.

Can you point me to where that would be modified? Would that be the plugin end?

tomfaulhaber commented 2 years ago

Yeah, you can change the code in upload_fileobj()(here) to specify whatever bucket and prefix you would like.