Merge branch 'main' into models/new_resnet_weights

Author: datumbox

.circleci/config.yml

@@ -190,7 +190,7 @@ jobs:
       - run:
           name: Required lint modifications
           when: on_fail
-          command: git diff
+          command: git --no-pager diff
 
   python_type_check:
     docker:
@@ -274,7 +274,7 @@ jobs:
           command: pip install --user --progress-bar off --no-build-isolation .
       - run:
           name: Install test utilities
-          command: pip install --user --progress-bar=off pytest
+          command: pip install --user --progress-bar=off pytest pytest-mock
       - run:
           name: Run tests
           command: pytest test/test_prototype_*.py

.circleci/config.yml.in

@@ -14,6 +14,7 @@ docs/build
 # sphinx-gallery
 docs/source/auto_examples/
 docs/source/gen_modules/
+docs/source/generated/
 # pytorch-sphinx-theme gets installed here
 docs/src
 

.gitignore

@@ -32,6 +32,7 @@ clean:
 	rm -rf $(BUILDDIR)/*
 	rm -rf $(SOURCEDIR)/auto_examples/  # sphinx-gallery
 	rm -rf $(SOURCEDIR)/gen_modules/  # sphinx-gallery
+	rm -rf $(SOURCEDIR)/generated/  # autosummary
 
 .PHONY: help Makefile docset
 

docs/Makefile

@@ -0,0 +1,9 @@
+.. role:: hidden
+    :class: hidden-section
+.. currentmodule:: {{ module }}
+
+
+{{ name | underline}}
+
+.. autoclass:: {{ name }}
+    :members:

docs/source/_templates/class.rst

@@ -0,0 +1,12 @@
+.. role:: hidden
+    :class: hidden-section
+.. currentmodule:: {{ module }}
+
+
+{{ name | underline}}
+
+.. autoclass:: {{ name }}
+    :members:
+        __getitem__,
+        {% if "category_name" in methods %} category_name {% endif %}
+    :special-members:

docs/source/_templates/class_dataset.rst

@@ -0,0 +1,8 @@
+.. role:: hidden
+    :class: hidden-section
+.. currentmodule:: {{ module }}
+
+
+{{ name | underline}}
+
+.. autofunction:: {{ name }}

docs/source/_templates/function.rst

@@ -144,6 +144,9 @@
 htmlhelp_basename = "PyTorchdoc"
 
 
+autosummary_generate = True
+
+
 # -- Options for LaTeX output ---------------------------------------------
 latex_elements = {
     # The paper size ('letterpaper' or 'a4paper').