wrap dtensor import for OSS compatibility #124

	name: Build and Update Docs

	on:
	push:
	branches: [ main ]

	# Allow one concurrent deployment
	concurrency:
	group: "pages"
	cancel-in-progress: true

	jobs:
	build_docs:
	runs-on: ubuntu-latest
	steps:
	- name: Check out repo
	uses: actions/checkout@v2
	- name: Setup conda env
	uses: conda-incubator/setup-miniconda@v2
	with:
	miniconda-version: "latest"
	activate-environment: test
	- name: Install dependencies
	shell: bash -l {0}
	run: \|
	set -eux
	conda activate test
	conda install pytorch cpuonly -c pytorch-nightly
	pip install -r requirements.txt
	pip install -r dev-requirements.txt
	python setup.py sdist bdist_wheel
	pip install dist/*.whl
	- name: Build docs
	shell: bash -l {0}
	run: \|
	set -eux
	conda activate test
	cd docs
	pip install -r requirements.txt
	sphinx-build -b html source build/html/main
	touch build/html/main/.nojekyll
	cd ..
	- name: Deploy docs to Github pages
	uses: JamesIves/[email protected]
	with:
	branch: gh-pages # The branch the action should deploy to.
	folder: docs/build/html/main # The folder the action should deploy.
	target-folder: main

Provide feedback