chore(archai): Adds updated files.

.amltignore

@@ -1,4 +1,7 @@
-# Folders to be ignored when using Amulet
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
+# Folders to be ignored when using AMLT
 tests/
 tools/
 archived/

.github/pull_request_template.md

@@ -0,0 +1,37 @@
+# Pull Request Template
+
+## Description
+
+Please include a list of changes, relevant information, context and new dependencies.
+
+Fixes: # (issue)
+
+## Changes
+
+Please delete options that are not relevant.
+
+- [ ] Bug fix (non-breaking change which fixes an issue)
+- [ ] New feature (non-breaking change which adds functionality)
+- [ ] Breaking change (causes existing functionality to not work as expected)
+- [ ] Documentation update
+
+## Tests
+
+Describe if additional tests have been implemented and reproduced.
+
+**Configuration**:
+* Archai version:
+* Platform:
+* Environment:
+
+## Final checks:
+
+- [ ] Follows guidelines of project
+- [ ] Self-review of code
+- [ ] Commented code in hard-to-understand areas
+- [ ] Corresponding changes to the documentation
+- [ ] Changes generate no new warnings
+- [ ] Added tests to assert changes
+- [ ] Unit tests pass locally with added changes
+- [ ] Dependent changes have already been merged
+- [ ] Checked code correction any misspellings

.github/workflows/build-publish-docs.yml

@@ -0,0 +1,41 @@
+name: Build and Publish Documentation
+
+on:
+  push:
+    branches:
+      - master
+  pull_request:
+    branches:
+      - master
+
+jobs:
+  build-publish:
+    name: Builds and publishes a Sphinx-based documentation
+    runs-on: ubuntu-latest
+    steps:
+      - name: Pulls the repository
+        uses: actions/checkout@v2
+        with:
+          fetch-depth: 0
+      - name: Sets up the conda environment
+        uses: conda-incubator/setup-miniconda@v2
+        with:
+          python-version: 3.7
+          activate-environment: archai
+      - name: Installs the requirements
+        shell: bash -l {0}
+        run: |
+          python -m pip install --user -e .
+          cd docs
+          python -m pip install --user -r requirements.txt
+      - name: Builds the documentation
+        shell: bash -l {0}
+        run: |
+          cd docs
+          make clean
+          make html
+      - name: Pushes to remote branch
+        uses: peaceiris/actions-gh-pages@v3
+        with:
+          github_token: ${{ secrets.GITHUB_TOKEN }}
+          publish_dir: docs/_build/html/

.github/workflows/build-release-pypi.yml

@@ -0,0 +1,42 @@
+name: Build and Release to PyPI
+
+on:
+  push:
+    tags:
+      - "**"
+
+jobs:
+  publish-pypi:
+    name: Publishes release candidate to PyPI
+    runs-on: ubuntu-latest
+    steps:
+      - name: Pulls the repository
+        uses: actions/checkout@v2
+        with:
+          fetch-depth: 0
+      - name: Sets the Environment
+        # Installs a Python 3.7 environment
+        uses: actions/setup-python@v1
+        with:
+          python-version: 3.7
+      - name: Installs pypa/build
+        # Install the `build` package for further building
+        run: >-
+          python -m
+          pip install
+          build
+          --user
+      - name: Builds the Wheel
+        # Runs the `build` command to create the wheel and source tarball
+        run: >-
+          python -m
+          build
+          --sdist
+          --wheel
+          --outdir dist/
+          .
+      - name: Publishes to PyPI
+        # Publishes the builded package to PyPI
+        uses: pypa/gh-action-pypi-publish@master
+        with:
+          password: ${{ secrets.PYPI_API_TOKEN }}

.gitignore

@@ -33,8 +33,8 @@ wheels/
 MANIFEST
 
 # PyInstaller
-#  Usually these files are written by a python script from a template
-#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+# Usually these files are written by a python script from a template
+# before PyInstaller builds the exe, so as to inject date/other infos into it.
 *.manifest
 *.spec
 

AUTHORS.md

@@ -1,4 +1,4 @@
 # Authors
 
 - [Shital Shah](http://www.shitalshah.com)
-- [Debadeepta Dey](https://www.debadeepta.com/)
+- [Debadeepta Dey](https://www.debadeepta.com)

LICENSE

@@ -1,5 +1,3 @@
-    Archai
-
     MIT License
 
     Copyright (c) Microsoft Corporation.

NOTICE.md

@@ -1,10 +1,10 @@
 NOTICES AND INFORMATION
 Do Not Translate or Localize
 
-This software incorporates material from third parties. Microsoft makes certain
-open source code available at http://3rdpartysource.microsoft.com, or you may
-send a check or money order for US $5.00, including the product name, the open
-source component name, and version number, to:
+This software incorporates material from third parties.
+Microsoft makes certain open source code available at https://3rdpartysource.microsoft.com,
+or you may send a check or money order for US $5.00, including the product name,
+the open source component name, platform, and version number, to:
 
 Source Code Compliance Team
 Microsoft Corporation
@@ -12,10 +12,10 @@ One Microsoft Way
 Redmond, WA 98052
 USA
 
-Notwithstanding any other terms, you may reverse engineer this software to the
-extent required to debug changes to any libraries licensed under the GNU Lesser
-General Public License.
+Notwithstanding any other terms, you may reverse engineer this software to the extent
+required to debug changes to any libraries licensed under the GNU Lesser General Public License.
 
+The following components are used within Archai and are presented along their license.
 
 **Component.** https://github.com/kakaobrain/fast-autoaugment
 

README.md

@@ -63,7 +63,7 @@ We would love community contributions, feedback, questions, algorithm implementa
 Join the Archai group on [Facebook](https://www.facebook.com/groups/1133660130366735/) to stay up to date or ask any questions.
 
 ## Team
-Archai has been created and maintained by [Shital Shah](https://shitalshah.com) and [Debadeepta Dey](www.debadeepta.com) in the [Reinforcement Learning Group](https://www.microsoft.com/en-us/research/group/reinforcement-learning-redmond/) at Microsoft Research AI, Redmond, USA. Archai has benefited immensely from discussions with [John Langford](https://www.microsoft.com/en-us/research/people/jcl/), [Rich Caruana](https://www.microsoft.com/en-us/research/people/rcaruana/), [Eric Horvitz](https://www.microsoft.com/en-us/research/people/horvitz/) and [Alekh Agarwal](https://www.microsoft.com/en-us/research/people/alekha/).
+Archai has been created and maintained by [Shital Shah](https://shitalshah.com), [Debadeepta Dey](www.debadeepta.com), [Gustavo de Rosa](https://www.microsoft.com/en-us/research/people/gderosa/), [Caio Cesar Teodoro Mendes]() in the [Reinforcement Learning Group](https://www.microsoft.com/en-us/research/group/reinforcement-learning-redmond/) at Microsoft Research, Redmond, USA. Archai has benefited immensely from discussions with [Ofer Dekel](https://www.microsoft.com/en-us/research/people/oferd/), [John Langford](https://www.microsoft.com/en-us/research/people/jcl/), [Rich Caruana](https://www.microsoft.com/en-us/research/people/rcaruana/), [Eric Horvitz](https://www.microsoft.com/en-us/research/people/horvitz/) and [Alekh Agarwal](https://www.microsoft.com/en-us/research/people/alekha/).
 
 We look forward to Archai becoming more community driven and including major contributors here.
 

azure-pipelines.yml

@@ -1,8 +1,5 @@
-# Python package
-
-# Create and test a Python package on multiple Python versions.
-# Add steps that analyze code, save the dist with the build record, publish to a PyPI-compatible index, and more:
-# https://docs.microsoft.com/azure/devops/pipelines/languages/python
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
 
 trigger:
 - master
@@ -11,8 +8,6 @@ pool:
   vmImage: 'windows-latest'
 strategy:
   matrix:
-    # Python36:
-    #   python.version: '3.6'
     Python37:
       python.version: '3.7'
 

docs/Makefile

@@ -0,0 +1,20 @@
+# Minimal makefile for Sphinx documentation
+#
+
+# You can set these variables from the command line, and also
+# from the environment for the first two.
+SPHINXOPTS    ?=
+SPHINXBUILD   ?= sphinx-build
+SOURCEDIR     = .
+BUILDDIR      = _build
+
+# Put it first so that "make" without argument is like "make help".
+help:
+	@$(SPHINXBUILD) -M help "$(SOURCEDIR)" "$(BUILDDIR)" $(SPHINXOPTS) $(O)
+
+.PHONY: help Makefile
+
+# Catch-all target: route all unknown targets to Sphinx using the new
+# "make mode" option.  $(O) is meant as a shortcut for $(SPHINXOPTS).
+%: Makefile
+	@$(SPHINXBUILD) -M $@ "$(SOURCEDIR)" "$(BUILDDIR)" $(SPHINXOPTS) $(O)

docs/conf.py

@@ -11,8 +11,7 @@
 # documentation root, use os.path.abspath to make it absolute, like shown here.
 #
 
-import recommonmark
-from recommonmark.transform import AutoStructify
+# from recommonmark.transform import AutoStructify
 
 import os
 import sys
@@ -22,7 +21,7 @@ sys.path.insert(0, os.path.abspath('../archai'))
 # -- Project information -----------------------------------------------------
 
 project = 'Archai'
-copyright = '2020, Microsoft'
+copyright = '2022, Microsoft'
 author = ''
 
 
@@ -32,8 +31,8 @@ author = ''
 # extensions coming with Sphinx (named 'sphinx.ext.*') or your custom
 # ones.
 extensions = [
-    'rinoh.frontend.sphinx', 'sphinx.ext.autodoc', 'sphinx.ext.todo',
-    'sphinx.ext.viewcode', 'recommonmark', 'sphinx_rtd_theme' #, 'sphinx.ext.autosectionlabel' #, 'm2r2'
+    'sphinx.ext.autodoc', 'sphinx.ext.todo',
+    'sphinx.ext.viewcode', 'recommonmark', 'sphinx_rtd_theme'
 ]
 
 # Add any paths that contain templates here, relative to this directory.
@@ -46,7 +45,7 @@ exclude_patterns = [
     #'.*/',
     'cifar10_models/**',
     'data_aug/**',
-    'netowrks/**',
+    'networks/**',
     # 'confs/',
     # 'dockers/',
     # 'models/',
@@ -61,17 +60,16 @@ exclude_patterns = [
 # The theme to use for HTML and HTML Help pages.  See the documentation for
 # a list of builtin themes.
 #
-import sphinx_rtd_theme
 html_theme = 'sphinx_rtd_theme' # 'klink' #'bizstyle' #'alabaster'
 
 # Add any paths that contain custom static files (such as style sheets) here,
 # relative to this directory. They are copied after the builtin static files,
 # so a file named "default.css" will overwrite the builtin "default.css".
-html_static_path = ['_static']
+# html_static_path = ['_static']
 
-def setup(app):
-    app.add_config_value('recommonmark_config', {
-            'url_resolver': lambda url: github_doc_root + url,
-            'auto_toc_tree_section': 'Contents',
-            }, True)
-    app.add_transform(AutoStructify)
+# def setup(app):
+#     app.add_config_value('recommonmark_config', {
+#             'url_resolver': lambda url: github_doc_root + url,
+#             'auto_toc_tree_section': 'Contents',
+#             }, True)
+#     app.add_transform(AutoStructify)

docs/make.bat

@@ -0,0 +1,35 @@
+@ECHO OFF
+
+pushd %~dp0
+
+REM Command file for Sphinx documentation
+
+if "%SPHINXBUILD%" == "" (
+	set SPHINXBUILD=sphinx-build
+)
+set SOURCEDIR=.
+set BUILDDIR=_build
+
+if "%1" == "" goto help
+
+%SPHINXBUILD% >NUL 2>NUL
+if errorlevel 9009 (
+	echo.
+	echo.The 'sphinx-build' command was not found. Make sure you have Sphinx
+	echo.installed, then set the SPHINXBUILD environment variable to point
+	echo.to the full path of the 'sphinx-build' executable. Alternatively you
+	echo.may add the Sphinx directory to PATH.
+	echo.
+	echo.If you don't have Sphinx installed, grab it from
+	echo.http://sphinx-doc.org/
+	exit /b 1
+)
+
+%SPHINXBUILD% -M %1 %SOURCEDIR% %BUILDDIR% %SPHINXOPTS% %O%
+goto end
+
+:help
+%SPHINXBUILD% -M help %SOURCEDIR% %BUILDDIR% %SPHINXOPTS% %O%
+
+:end
+popd 

docs/requirements.txt

@@ -0,0 +1,3 @@
+recommonmark>=0.7.1
+sphinx>=4.1.2
+sphinx-rtd-theme>=0.5.2

install.bat

@@ -1,4 +1,7 @@
-REM below will be required for Python 3.7 and below if pickle wasn't updated
+REM Copyright (c) Microsoft Corporation.
+REM Licensed under the MIT license.
+
+REM Below will be required for Python 3.7 and below if pickle wasn't updated
 REM otherwise ray.init() fails if below is not done
 REM CALL conda install -y -c conda-forge pickle5
 

install.sh

@@ -1,9 +1,13 @@
 #!/bin/bash
-#fail if any errors
+
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
+# Fails if any errors
 set -e
 set -o xtrace
 
-# below will be required for Python 3.7 and below if pickle wasn't updated
+# Below will be required for Python 3.7 and below if pickle wasn't updated
 # otherwise ray.init() fails if below is not done
 # conda install -y -c conda-forge pickle5
 

setup.cfg

@@ -0,0 +1,2 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.

setup.py

@@ -1,38 +1,45 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT license.
 
-import setuptools
+from setuptools import find_packages, setup
 
-with open('README.md', 'r', encoding='utf_8') as fh:
+with open("README.md", "r", encoding='utf_8') as fh:
     long_description = fh.read()
 
-install_requires = [
-    'pystopwatch2 @ git+https://github.com/ildoonet/pystopwatch2.git',
+install_requires=[
+    'pystopwatch2',
     'hyperopt',
     'tensorwatch>=0.9.1', 'tensorboard',
     'pretrainedmodels', 'tqdm', 'sklearn', 'matplotlib', 'psutil',
     'requests', 'seaborn', 'h5py', 'rarfile',
-    'gorilla', 'pyyaml', 'overrides<4.0.0', 'runstats', 'psutil', 'statopt',
-    'pyunpack', 'patool', 'ray>=1.0.0', 'Send2Trash'
+    'gorilla', 'pyyaml', 'overrides==3.1.0', 'runstats', 'psutil', 'statopt',
+    'pyunpack', 'patool', 'ray>=1.0.0', 'Send2Trash',
+    'transformers', 'pytorch_lightning', 'tokenizers', 'datasets',
+    'ftfy', # needed for text scoring, fixes text for you
+    # nvidia transformer-xl
+    'nv_dllogger', # needed for compatibility between pip and setuptools
+    'pytorch-transformers', 'sacremoses', 'pynvml'
 ]
 
-setuptools.setup(
-    name='archai',
-    version='0.5.2',
-    author='Shital Shah, Debadeepta Dey,',
-    author_email='shitals@microsoft.com, dedey@microsoft.com',
-    description='Research platform for Neural Architecture Search',
+setup(
+    name="archai",
+    version="0.6.4",
+    description="Research platform for Neural Architecture Search",
     long_description=long_description,
-    long_description_content_type='text/markdown',
-    url='https://github.com/microsoft/archai',
-    packages=setuptools.find_packages(),
-        license='MIT',
-    classifiers=(
-        'Programming Language :: Python :: 3',
-        'License :: OSI Approved :: MIT License',
-        'Operating System :: OS Independent',
-        'Intended Audience :: Science/Research',
-    ),
+    long_description_content_type="text/markdown",
+    author="Shital Shah, Debadeepta Dey",
+    author_email="shitals@microsoft.com, dedey@microsoft.com",
+    url="https://github.com/microsoft/archai",
+	license="MIT",
+    install_requires=install_requires,
+    packages=find_packages(),
     include_package_data=True,
-    install_requires=install_requires
+    classifiers=[
+        "Development Status :: 5 - Production/Stable",
+        "Intended Audience :: Developers",
+        "Intended Audience :: Science/Research",
+        "License :: OSI Approved :: MIT License",
+        "Topic :: Software Development :: Libraries",
+        "Topic :: Software Development :: Libraries :: Python Modules",
+    ],
 )