Updating training script in basic_interactive notebook

[Fiona-Waters]

Issue link

RHOAIENG-6475

What changes have been made

When running GPU utilising workloads using the basic_interactive demo notebook I encountered a lot of errors while running the training script provided in the notebook. As it was added more than a year ago and may be out of date I have updated the notebook to use ray train using these examples: Text classification, Hugging Face Transformers + Ray Train

I have also updated the hf_interactive notebook which was using the same training script.

Verification steps

• Install RHOAI with the CodeFlare components set to Managed
• Ensure that your cluster has GPU nodes as required.
• Within a data science project workbench clone this branch and install codeflare sdk

git clone https://github.com/Fiona-Waters/codeflare-sdk.git -b udpate-notebook-script
cd codeflare-sdk
pip install poetry
poetry install

• Run the basic_interactive notebook and ensure that it works as expected.
• Note that aws credentials are required and you will need to create and use an s3 bucket while running the training script. See this PR for instructions.

Checks

• I've made sure the tests are passing.
• Testing Strategy
  • Unit tests
  • Manual tests
  • Testing is not required for this change

[Bobbins228]

I had to test with the latest-py39-cu118 image. It worked like a charm.
Happy to lgtm once the images are set back.
Awesome work Fiona 🚀

[Bobbins228]

/lgtm

[astefanutti]

/retest

[Bobbins228]

This PR just needs a rebase and it should pass the e2e test

[astefanutti]

/lgtm

[astefanutti]

/approve

[openshift-ci]

[APPROVALNOTIFIER] This PR is APPROVED

This pull-request has been approved by: astefanutti, Bobbins228

The full list of commands accepted by this bot can be found here.

The pull request process is described here

Needs approval from an approver in each of these files:

• OWNERS [Bobbins228,astefanutti]

Approvers can indicate their approval by writing /approve in a comment
Approvers can cancel approval by writing /approve cancel in a comment