Update Notebook Image to sdk0.3.1 and to use torchx from codeflare org (#65)

* update torchx to codeflare org

* update notebook image to sdk0.3.1-v1

Author: MichaelClifford

custom-nb-image/Dockerfile

@@ -23,8 +23,8 @@ RUN pip install -r requirements.txt
 RUN pip uninstall pickle5 -y 
 
 # Install codeflare-sdk and other libraries
-RUN pip install codeflare-sdk==0.3.0 \
+RUN pip install codeflare-sdk==0.3.1 \
 				datasets==2.6.1 \
 				transformers==4.23.1 \
 				evaluate==0.3.0 \
-				git+https://github.com/MichaelClifford/torchx.git@ray2-patch
+				git+https://github.com/project-codeflare/torchx.git@OCP

custom-nb-image/VERSION

@@ -1 +1 @@
-sdk0.3.0-v1
+sdk0.3.1-v1