Improve TensorFlow scraper

assets/javascripts/views/pages/simple.coffee

@@ -12,6 +12,7 @@ app.views.MeteorPage =
 app.views.RamdaPage =
 app.views.ReactPage =
 app.views.RethinkdbPage =
+app.views.TensorflowPage =
 app.views.TypescriptPage =
 app.views.UnderscorePage =
 app.views.SimplePage

assets/javascripts/views/pages/tensorflow.coffee

@@ -1,7 +0,0 @@
-#= require views/pages/base
-
-class app.views.TensorflowPage extends app.views.BasePage
-  prepare: ->
-    @highlightCode @findAll('pre[class*="lang-c++"]'), 'cpp'
-    @highlightCode @findAll('pre.lang-python'), 'python'
-    return

lib/docs/filters/tensorflow/clean_html.rb

@@ -9,6 +9,16 @@ module Docs
           node.parent.content = node.content
         end
 
+        css('pre').each do |node|
+          node.inner_html = node.inner_html.strip_heredoc
+
+          if node['class'].include?('lang-c++')
+            node['data-language'] = 'cpp'
+          elsif node['class'].include?('lang-python')
+            node['data-language'] = 'python'
+          end
+        end
+
         css('b').each do |node|
           node.before(node.children).remove
         end