docs: fixups for notebook + badge (#91)

Summary:
Some small docs fixups:

* install torchx with kfp for notebooks
* reduces spam + make extraction more flexible
* print the pipeline.yaml at end of example
* added shield for pypi version to README

Pull Request resolved: #91

Test Plan:
CI

https://colab.research.google.com/github/pytorch/torchx/blob/gh-pages/0.1.0.dev2/_downloads/1aa1c6eb4233b94bdb0a2d1581cc0b03/advanced_pipeline.ipynb

![2021-06-24-155948_823x625_scrot](https://user-images.githubusercontent.com/909104/123342753-36732280-d505-11eb-86d1-cde2f4157ef6.png)

https://github.com/pytorch/torchx/blob/11fd5db8aedba423b30d27ce756477cc5bf312f1/README.md

Reviewed By: kiukchung

Differential Revision: D29377781

Pulled By: d4l3k

fbshipit-source-id: 1933393a2ded461105737f900f2d385ab280557f

Author: d4l3k

README.md

@@ -1,4 +1,4 @@
-[![License](https://img.shields.io/badge/License-BSD%203--Clause-blue.svg)](LICENSE)![Tests](https://github.com/pytorch/torchx/actions/workflows/python-unittests.yaml/badge.svg)![Lint](https://github.com/pytorch/torchx/actions/workflows/lint.yaml/badge.svg)
+[![PyPI](https://img.shields.io/pypi/v/torchx)](https://pypi.org/project/torchx/) [![License](https://img.shields.io/badge/License-BSD%203--Clause-blue.svg)](LICENSE) ![Tests](https://github.com/pytorch/torchx/actions/workflows/python-unittests.yaml/badge.svg) ![Lint](https://github.com/pytorch/torchx/actions/workflows/lint.yaml/badge.svg)
 
 
 # TorchX

docs/source/conf.py

@@ -312,9 +312,9 @@ def handle_item(fieldarg, content):
     code_url = "https://github.com/pytorch/torchx/archive/refs/heads/master.tar.gz"
 
 first_notebook_cell = f"""
-!pip install torchx
+!pip install torchx[kfp]
 !wget --no-clobber {code_url}
-!tar xvf {notebook_version}.tar.gz torchx-{notebook_version} --strip-components=1
+!tar xf {notebook_version}.tar.gz --strip-components=1
 
 NOTEBOOK = True
 """.strip()

examples/pipelines/kfp/advanced_pipeline.py

@@ -267,6 +267,9 @@ def pipeline() -> None:
     package_path=args.package_path,
 )
 
+with open(args.package_path, "rt") as f:
+    print(f.read())
+
 # %%
 # Once this has all run you should have a pipeline file (typically
 # pipeline.yaml) that you can upload to your KFP cluster via the UI or

examples/pipelines/kfp/intro_pipeline.py

@@ -59,6 +59,9 @@ def pipeline() -> None:
     package_path="pipeline.yaml",
 )
 
+with open("pipeline.yaml", "rt") as f:
+    print(f.read())
+
 # %%
 # Once this has all run you should have a pipeline file (typically
 # pipeline.yaml) that you can upload to your KFP cluster via the UI or