Skip to content

Commit

Permalink
Merge pull request freeCodeCamp#1241 from xavieryao/master
Browse files Browse the repository at this point in the history
Add PyTorch v1.7.0 documentation
  • Loading branch information
simon04 authored Nov 14, 2020
2 parents b216b4d + cc41db8 commit 41ab102
Show file tree
Hide file tree
Showing 7 changed files with 104 additions and 0 deletions.
5 changes: 5 additions & 0 deletions assets/javascripts/templates/pages/about_tmpl.coffee
Original file line number Diff line number Diff line change
Expand Up @@ -626,6 +626,11 @@ credits = [
'2001-2020 Python Software Foundation<br>Python is a trademark of the Python Software Foundation.',
'PSFL',
'https://docs.python.org/3/license.html'
], [
'PyTorch',
'2019 Torch Contributors',
'BSD',
'https://raw.githubusercontent.com/pytorch/pytorch/master/LICENSE'
], [
'Q',
'2009-2017 Kristopher Michael Kowal',
Expand Down
12 changes: 12 additions & 0 deletions lib/docs/filters/pytorch/clean_html.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
module Docs
class Pytorch
class CleanHtmlFilter < Filter
def call
@doc = at_css('.pytorch-article')
# Show katex-mathml nodes and remove katex-html nodes
css('.katex-html').remove
doc
end
end
end
end
54 changes: 54 additions & 0 deletions lib/docs/filters/pytorch/entries.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,54 @@
module Docs
class Pytorch
class EntriesFilter < Docs::EntriesFilter
def get_name
# The id of the container `div.section` indicates the page type.
# If the id starts with `module-`, then it's an API reference,
# otherwise it is a note or design doc.
# After the `sphinx/clean_html` filter, that id is assigned to the second element.
if doc.element_children[1]['id']&.starts_with? 'module-'
/\Amodule-(.*)/.match(doc.element_children[1]['id'])[1]
else
at_css('h1').content
end
end

def get_type
name
end

def include_default_entry?
# If the page is not an API reference, we only include it in the index when it
# contains additional entries. See the doc for `get_name`.
doc.element_children[1]['id']&.starts_with? 'module-'
end

def additional_entries
return [] if root_page?

entries = []

css('dt').each do |node|
name = node['id']
if name == self.name or name == nil
next
end

case node.parent['class']
when 'method', 'function'
if node.at_css('code').content.starts_with? 'property '
# this instance method is a property, so treat it as an attribute
entries << [name, node['id']]
else
entries << [name + '()', node['id']]
end
when 'class', 'attribute'
entries << [name, node['id']]
end
end

entries
end
end
end
end
32 changes: 32 additions & 0 deletions lib/docs/scrapers/pytorch.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,32 @@
module Docs
class Pytorch < UrlScraper
self.name = 'PyTorch'
self.slug = 'pytorch'
self.type = 'sphinx'
self.force_gzip = true
self.links = {
home: 'https://pytorch.org/',
code: 'https://github.com/pytorch/pytorch'
}

html_filters.push 'pytorch/clean_html', 'sphinx/clean_html', 'pytorch/entries'

options[:skip] = ['cpp_index.html', 'packages.html', 'py-modindex.html', 'genindex.html']
options[:skip_patterns] = [/\Acommunity/, /\A_modules/, /\Anotes/, /\Aorg\/pytorch\//]
options[:max_image_size] = 256_000

options[:attribution] = <<-HTML
&copy; 2019 Torch Contributors<br>
Licensed under the 3-clause BSD License.
HTML

version do
self.release = '1.7.0'
self.base_url = "https://pytorch.org/docs/#{release}/"
end

def get_latest_version(opts)
get_latest_github_release('pytorch', 'pytorch', opts)
end
end
end
Binary file added public/icons/docs/pytorch/16.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added public/icons/docs/pytorch/16@2x.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
1 change: 1 addition & 0 deletions public/icons/docs/pytorch/SOURCE
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
https://pytorch.org/favicon.ico

0 comments on commit 41ab102

Please sign in to comment.