http://git-wip-us.apache.org/repos/asf/incubator-airflow-site/blob/1f06fa0e/_modules/airflow/contrib/operators/vertica_to_hive.html
----------------------------------------------------------------------
diff --git a/_modules/airflow/contrib/operators/vertica_to_hive.html 
b/_modules/airflow/contrib/operators/vertica_to_hive.html
index 6e6f8c4..24b843e 100644
--- a/_modules/airflow/contrib/operators/vertica_to_hive.html
+++ b/_modules/airflow/contrib/operators/vertica_to_hive.html
@@ -268,7 +268,7 @@
         <span class="p">}</span>
         <span class="k">return</span> <span class="n">d</span><span 
class="p">[</span><span class="n">vertica_type</span><span class="p">]</span> 
<span class="k">if</span> <span class="n">vertica_type</span> <span 
class="ow">in</span> <span class="n">d</span> <span class="k">else</span> <span 
class="s1">&#39;STRING&#39;</span>
 
-    <span class="k">def</span> <span class="nf">execute</span><span 
class="p">(</span><span class="bp">self</span><span class="p">,</span> <span 
class="n">context</span><span class="p">):</span>
+<div class="viewcode-block" id="VerticaToHiveTransfer.execute"><a 
class="viewcode-back" 
href="../../../../code.html#airflow.contrib.operators.vertica_to_hive.VerticaToHiveTransfer.execute">[docs]</a>
    <span class="k">def</span> <span class="nf">execute</span><span 
class="p">(</span><span class="bp">self</span><span class="p">,</span> <span 
class="n">context</span><span class="p">):</span>
         <span class="n">hive</span> <span class="o">=</span> <span 
class="n">HiveCliHook</span><span class="p">(</span><span 
class="n">hive_cli_conn_id</span><span class="o">=</span><span 
class="bp">self</span><span class="o">.</span><span 
class="n">hive_cli_conn_id</span><span class="p">)</span>
         <span class="n">vertica</span> <span class="o">=</span> <span 
class="n">VerticaHook</span><span class="p">(</span><span 
class="n">vertica_conn_id</span><span class="o">=</span><span 
class="bp">self</span><span class="o">.</span><span 
class="n">vertica_conn_id</span><span class="p">)</span>
 
@@ -297,7 +297,7 @@
                 <span class="n">create</span><span class="o">=</span><span 
class="bp">self</span><span class="o">.</span><span 
class="n">create</span><span class="p">,</span>
                 <span class="n">partition</span><span class="o">=</span><span 
class="bp">self</span><span class="o">.</span><span 
class="n">partition</span><span class="p">,</span>
                 <span class="n">delimiter</span><span class="o">=</span><span 
class="bp">self</span><span class="o">.</span><span 
class="n">delimiter</span><span class="p">,</span>
-                <span class="n">recreate</span><span class="o">=</span><span 
class="bp">self</span><span class="o">.</span><span 
class="n">recreate</span><span class="p">)</span></div>
+                <span class="n">recreate</span><span class="o">=</span><span 
class="bp">self</span><span class="o">.</span><span 
class="n">recreate</span><span class="p">)</span></div></div>
 </pre></div>
 
            </div>
@@ -328,20 +328,13 @@
 
   
 
-    <script type="text/javascript">
-        var DOCUMENTATION_OPTIONS = {
-            URL_ROOT:'../../../../',
-            VERSION:'',
-            LANGUAGE:'None',
-            COLLAPSE_INDEX:false,
-            FILE_SUFFIX:'.html',
-            HAS_SOURCE:  true,
-            SOURCELINK_SUFFIX: '.txt'
-        };
-    </script>
-      <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
+    
+      <script type="text/javascript" id="documentation_options" 
data-url_root="../../../../" 
src="../../../../_static/documentation_options.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
 
   
 

http://git-wip-us.apache.org/repos/asf/incubator-airflow-site/blob/1f06fa0e/_modules/airflow/contrib/operators/winrm_operator.html
----------------------------------------------------------------------
diff --git a/_modules/airflow/contrib/operators/winrm_operator.html 
b/_modules/airflow/contrib/operators/winrm_operator.html
index fa546c3..ec9a1f2 100644
--- a/_modules/airflow/contrib/operators/winrm_operator.html
+++ b/_modules/airflow/contrib/operators/winrm_operator.html
@@ -223,7 +223,7 @@
         <span class="bp">self</span><span class="o">.</span><span 
class="n">timeout</span> <span class="o">=</span> <span class="n">timeout</span>
         <span class="bp">self</span><span class="o">.</span><span 
class="n">do_xcom_push</span> <span class="o">=</span> <span 
class="n">do_xcom_push</span>
 
-    <span class="k">def</span> <span class="nf">execute</span><span 
class="p">(</span><span class="bp">self</span><span class="p">,</span> <span 
class="n">context</span><span class="p">):</span>
+<div class="viewcode-block" id="WinRMOperator.execute"><a 
class="viewcode-back" 
href="../../../../code.html#airflow.contrib.operators.winrm_operator.WinRMOperator.execute">[docs]</a>
    <span class="k">def</span> <span class="nf">execute</span><span 
class="p">(</span><span class="bp">self</span><span class="p">,</span> <span 
class="n">context</span><span class="p">):</span>
         <span class="k">try</span><span class="p">:</span>
             <span class="k">if</span> <span class="bp">self</span><span 
class="o">.</span><span class="n">ssh_conn_id</span> <span 
class="ow">and</span> <span class="ow">not</span> <span 
class="bp">self</span><span class="o">.</span><span 
class="n">winrm_hook</span><span class="p">:</span>
                 <span class="bp">self</span><span class="o">.</span><span 
class="n">log</span><span class="o">.</span><span class="n">info</span><span 
class="p">(</span><span class="s2">&quot;hook not found, 
creating&quot;</span><span class="p">)</span>
@@ -268,7 +268,7 @@
         <span class="k">except</span> <span class="ne">Exception</span> <span 
class="k">as</span> <span class="n">e</span><span class="p">:</span>
             <span class="k">raise</span> <span 
class="n">AirflowException</span><span class="p">(</span><span 
class="s2">&quot;WinRM operator error: </span><span class="si">{0}</span><span 
class="s2">&quot;</span><span class="o">.</span><span 
class="n">format</span><span class="p">(</span><span class="nb">str</span><span 
class="p">(</span><span class="n">e</span><span class="p">)))</span>
 
-        <span class="k">return</span> <span class="kc">True</span></div>
+        <span class="k">return</span> <span class="kc">True</span></div></div>
 </pre></div>
 
            </div>
@@ -299,20 +299,13 @@
 
   
 
-    <script type="text/javascript">
-        var DOCUMENTATION_OPTIONS = {
-            URL_ROOT:'../../../../',
-            VERSION:'',
-            LANGUAGE:'None',
-            COLLAPSE_INDEX:false,
-            FILE_SUFFIX:'.html',
-            HAS_SOURCE:  true,
-            SOURCELINK_SUFFIX: '.txt'
-        };
-    </script>
-      <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
+    
+      <script type="text/javascript" id="documentation_options" 
data-url_root="../../../../" 
src="../../../../_static/documentation_options.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
 
   
 

http://git-wip-us.apache.org/repos/asf/incubator-airflow-site/blob/1f06fa0e/_modules/airflow/contrib/sensors/aws_redshift_cluster_sensor.html
----------------------------------------------------------------------
diff --git a/_modules/airflow/contrib/sensors/aws_redshift_cluster_sensor.html 
b/_modules/airflow/contrib/sensors/aws_redshift_cluster_sensor.html
index 84012a0..55215d1 100644
--- a/_modules/airflow/contrib/sensors/aws_redshift_cluster_sensor.html
+++ b/_modules/airflow/contrib/sensors/aws_redshift_cluster_sensor.html
@@ -241,20 +241,13 @@
 
   
 
-    <script type="text/javascript">
-        var DOCUMENTATION_OPTIONS = {
-            URL_ROOT:'../../../../',
-            VERSION:'',
-            LANGUAGE:'None',
-            COLLAPSE_INDEX:false,
-            FILE_SUFFIX:'.html',
-            HAS_SOURCE:  true,
-            SOURCELINK_SUFFIX: '.txt'
-        };
-    </script>
-      <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
+    
+      <script type="text/javascript" id="documentation_options" 
data-url_root="../../../../" 
src="../../../../_static/documentation_options.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
 
   
 

http://git-wip-us.apache.org/repos/asf/incubator-airflow-site/blob/1f06fa0e/_modules/airflow/contrib/sensors/bash_sensor.html
----------------------------------------------------------------------
diff --git a/_modules/airflow/contrib/sensors/bash_sensor.html 
b/_modules/airflow/contrib/sensors/bash_sensor.html
index f9858d8..b3c1619 100644
--- a/_modules/airflow/contrib/sensors/bash_sensor.html
+++ b/_modules/airflow/contrib/sensors/bash_sensor.html
@@ -284,20 +284,13 @@
 
   
 
-    <script type="text/javascript">
-        var DOCUMENTATION_OPTIONS = {
-            URL_ROOT:'../../../../',
-            VERSION:'',
-            LANGUAGE:'None',
-            COLLAPSE_INDEX:false,
-            FILE_SUFFIX:'.html',
-            HAS_SOURCE:  true,
-            SOURCELINK_SUFFIX: '.txt'
-        };
-    </script>
-      <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
+    
+      <script type="text/javascript" id="documentation_options" 
data-url_root="../../../../" 
src="../../../../_static/documentation_options.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
 
   
 

http://git-wip-us.apache.org/repos/asf/incubator-airflow-site/blob/1f06fa0e/_modules/airflow/contrib/sensors/bigquery_sensor.html
----------------------------------------------------------------------
diff --git a/_modules/airflow/contrib/sensors/bigquery_sensor.html 
b/_modules/airflow/contrib/sensors/bigquery_sensor.html
index aa38b78..31f2b43 100644
--- a/_modules/airflow/contrib/sensors/bigquery_sensor.html
+++ b/_modules/airflow/contrib/sensors/bigquery_sensor.html
@@ -259,20 +259,13 @@
 
   
 
-    <script type="text/javascript">
-        var DOCUMENTATION_OPTIONS = {
-            URL_ROOT:'../../../../',
-            VERSION:'',
-            LANGUAGE:'None',
-            COLLAPSE_INDEX:false,
-            FILE_SUFFIX:'.html',
-            HAS_SOURCE:  true,
-            SOURCELINK_SUFFIX: '.txt'
-        };
-    </script>
-      <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
+    
+      <script type="text/javascript" id="documentation_options" 
data-url_root="../../../../" 
src="../../../../_static/documentation_options.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
 
   
 

http://git-wip-us.apache.org/repos/asf/incubator-airflow-site/blob/1f06fa0e/_modules/airflow/contrib/sensors/cassandra_sensor.html
----------------------------------------------------------------------
diff --git a/_modules/airflow/contrib/sensors/cassandra_sensor.html 
b/_modules/airflow/contrib/sensors/cassandra_sensor.html
new file mode 100644
index 0000000..0bafccf
--- /dev/null
+++ b/_modules/airflow/contrib/sensors/cassandra_sensor.html
@@ -0,0 +1,271 @@
+
+
+<!DOCTYPE html>
+<!--[if IE 8]><html class="no-js lt-ie9" lang="en" > <![endif]-->
+<!--[if gt IE 8]><!--> <html class="no-js" lang="en" > <!--<![endif]-->
+<head>
+  <meta charset="utf-8">
+  
+  <meta name="viewport" content="width=device-width, initial-scale=1.0">
+  
+  <title>airflow.contrib.sensors.cassandra_sensor &mdash; Airflow 
Documentation</title>
+  
+
+  
+  
+  
+  
+
+  
+
+  
+  
+    
+
+  
+
+  <link rel="stylesheet" href="../../../../_static/css/theme.css" 
type="text/css" />
+  <link rel="stylesheet" href="../../../../_static/pygments.css" 
type="text/css" />
+    <link rel="index" title="Index" href="../../../../genindex.html" />
+    <link rel="search" title="Search" href="../../../../search.html" /> 
+
+  
+  <script src="../../../../_static/js/modernizr.min.js"></script>
+
+</head>
+
+<body class="wy-body-for-nav">
+
+   
+  <div class="wy-grid-for-nav">
+
+    
+    <nav data-toggle="wy-nav-shift" class="wy-nav-side">
+      <div class="wy-side-scroll">
+        <div class="wy-side-nav-search">
+          
+
+          
+            <a href="../../../../index.html" class="icon icon-home"> Airflow
+          
+
+          
+          </a>
+
+          
+            
+            
+          
+
+          
+<div role="search">
+  <form id="rtd-search-form" class="wy-form" action="../../../../search.html" 
method="get">
+    <input type="text" name="q" placeholder="Search docs" />
+    <input type="hidden" name="check_keywords" value="yes" />
+    <input type="hidden" name="area" value="default" />
+  </form>
+</div>
+
+          
+        </div>
+
+        <div class="wy-menu wy-menu-vertical" data-spy="affix" 
role="navigation" aria-label="main navigation">
+          
+            
+            
+              
+            
+            
+              <ul>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../project.html">Project</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../license.html">License</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../start.html">Quick Start</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../installation.html">Installation</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../tutorial.html">Tutorial</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../howto/index.html">How-to Guides</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../ui.html">UI / Screenshots</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../concepts.html">Concepts</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../profiling.html">Data Profiling</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../cli.html">Command Line Interface</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../scheduler.html">Scheduling &amp; Triggers</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../plugins.html">Plugins</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../security.html">Security</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../timezone.html">Time zones</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../api.html">Experimental Rest API</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../integration.html">Integration</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../lineage.html">Lineage</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../faq.html">FAQ</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../code.html">API Reference</a></li>
+</ul>
+
+            
+          
+        </div>
+      </div>
+    </nav>
+
+    <section data-toggle="wy-nav-shift" class="wy-nav-content-wrap">
+
+      
+      <nav class="wy-nav-top" aria-label="top navigation">
+        
+          <i data-toggle="wy-nav-top" class="fa fa-bars"></i>
+          <a href="../../../../index.html">Airflow</a>
+        
+      </nav>
+
+
+      <div class="wy-nav-content">
+        
+        <div class="rst-content">
+        
+          
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+<div role="navigation" aria-label="breadcrumbs navigation">
+
+  <ul class="wy-breadcrumbs">
+    
+      <li><a href="../../../../index.html">Docs</a> &raquo;</li>
+        
+          <li><a href="../../../index.html">Module code</a> &raquo;</li>
+        
+      <li>airflow.contrib.sensors.cassandra_sensor</li>
+    
+    
+      <li class="wy-breadcrumbs-aside">
+        
+      </li>
+    
+  </ul>
+
+  
+  <hr/>
+</div>
+          <div role="main" class="document" itemscope="itemscope" 
itemtype="http://schema.org/Article";>
+           <div itemprop="articleBody">
+            
+  <h1>Source code for airflow.contrib.sensors.cassandra_sensor</h1><div 
class="highlight"><pre>
+<span></span><span class="c1"># -*- coding: utf-8 -*-</span>
+<span class="c1">#</span>
+<span class="c1"># Licensed to the Apache Software Foundation (ASF) under 
one</span>
+<span class="c1"># or more contributor license agreements.  See the NOTICE 
file</span>
+<span class="c1"># distributed with this work for additional information</span>
+<span class="c1"># regarding copyright ownership.  The ASF licenses this 
file</span>
+<span class="c1"># to you under the Apache License, Version 2.0 (the</span>
+<span class="c1"># &quot;License&quot;); you may not use this file except in 
compliance</span>
+<span class="c1"># with the License.  You may obtain a copy of the License 
at</span>
+<span class="c1">#</span>
+<span class="c1">#   http://www.apache.org/licenses/LICENSE-2.0</span>
+<span class="c1">#</span>
+<span class="c1"># Unless required by applicable law or agreed to in 
writing,</span>
+<span class="c1"># software distributed under the License is distributed on 
an</span>
+<span class="c1"># &quot;AS IS&quot; BASIS, WITHOUT WARRANTIES OR CONDITIONS 
OF ANY</span>
+<span class="c1"># KIND, either express or implied.  See the License for 
the</span>
+<span class="c1"># specific language governing permissions and 
limitations</span>
+<span class="c1"># under the License.</span>
+<span class="kn">from</span> <span 
class="nn">airflow.contrib.hooks.cassandra_hook</span> <span 
class="k">import</span> <span class="n">CassandraHook</span>
+<span class="kn">from</span> <span 
class="nn">airflow.sensors.base_sensor_operator</span> <span 
class="k">import</span> <span class="n">BaseSensorOperator</span>
+<span class="kn">from</span> <span class="nn">airflow.utils.decorators</span> 
<span class="k">import</span> <span class="n">apply_defaults</span>
+
+
+<div class="viewcode-block" id="CassandraRecordSensor"><a 
class="viewcode-back" 
href="../../../../code.html#airflow.contrib.sensors.cassandra_sensor.CassandraRecordSensor">[docs]</a><span
 class="k">class</span> <span class="nc">CassandraRecordSensor</span><span 
class="p">(</span><span class="n">BaseSensorOperator</span><span 
class="p">):</span>
+    <span class="sd">&quot;&quot;&quot;</span>
+<span class="sd">    Checks for the existence of a record in a Cassandra 
cluster.</span>
+
+<span class="sd">    For example, if you want to wait for a record that has 
values &#39;v1&#39; and &#39;v2&#39; for each</span>
+<span class="sd">    primary keys &#39;p1&#39; and &#39;p2&#39; to be 
populated in keyspace &#39;k&#39; and table &#39;t&#39;,</span>
+<span class="sd">    instantiate it as follows:</span>
+
+<span class="sd">    &gt;&gt;&gt; CassandraRecordSensor(table=&quot;k.t&quot;, 
keys={&quot;p1&quot;: &quot;v1&quot;, &quot;p2&quot;: &quot;v2&quot;},</span>
+<span class="sd">    ...     cassandra_conn_id=&quot;cassandra_default&quot;, 
task_id=&quot;cassandra_sensor&quot;)</span>
+<span class="sd">    &lt;Task(CassandraRecordSensor): 
cassandra_sensor&gt;</span>
+<span class="sd">    &quot;&quot;&quot;</span>
+    <span class="n">template_fields</span> <span class="o">=</span> <span 
class="p">(</span><span class="s1">&#39;table&#39;</span><span 
class="p">,</span> <span class="s1">&#39;keys&#39;</span><span 
class="p">)</span>
+
+    <span class="nd">@apply_defaults</span>
+    <span class="k">def</span> <span class="nf">__init__</span><span 
class="p">(</span><span class="bp">self</span><span class="p">,</span> <span 
class="n">table</span><span class="p">,</span> <span class="n">keys</span><span 
class="p">,</span> <span class="n">cassandra_conn_id</span><span 
class="p">,</span> <span class="o">*</span><span class="n">args</span><span 
class="p">,</span> <span class="o">**</span><span class="n">kwargs</span><span 
class="p">):</span>
+        <span class="sd">&quot;&quot;&quot;</span>
+<span class="sd">        Create a new CassandraRecordSensor</span>
+
+<span class="sd">        :param table: Target Cassandra table.</span>
+<span class="sd">                      Use dot notation to target a specific 
keyspace.</span>
+<span class="sd">        :type table: string</span>
+<span class="sd">        :param keys: The keys and their values to be 
monitored</span>
+<span class="sd">        :type keys: dict</span>
+<span class="sd">        :param cassandra_conn_id: The connection ID to 
use</span>
+<span class="sd">                                  when connecting to 
Cassandra cluster</span>
+<span class="sd">        :type cassandra_conn_id: string</span>
+<span class="sd">        &quot;&quot;&quot;</span>
+        <span class="nb">super</span><span class="p">(</span><span 
class="n">CassandraRecordSensor</span><span class="p">,</span> <span 
class="bp">self</span><span class="p">)</span><span class="o">.</span><span 
class="fm">__init__</span><span class="p">(</span><span class="o">*</span><span 
class="n">args</span><span class="p">,</span> <span class="o">**</span><span 
class="n">kwargs</span><span class="p">)</span>
+        <span class="bp">self</span><span class="o">.</span><span 
class="n">cassandra_conn_id</span> <span class="o">=</span> <span 
class="n">cassandra_conn_id</span>
+        <span class="bp">self</span><span class="o">.</span><span 
class="n">table</span> <span class="o">=</span> <span class="n">table</span>
+        <span class="bp">self</span><span class="o">.</span><span 
class="n">keys</span> <span class="o">=</span> <span class="n">keys</span>
+
+<div class="viewcode-block" id="CassandraRecordSensor.poke"><a 
class="viewcode-back" 
href="../../../../code.html#airflow.contrib.sensors.cassandra_sensor.CassandraRecordSensor.poke">[docs]</a>
    <span class="k">def</span> <span class="nf">poke</span><span 
class="p">(</span><span class="bp">self</span><span class="p">,</span> <span 
class="n">context</span><span class="p">):</span>
+        <span class="bp">self</span><span class="o">.</span><span 
class="n">log</span><span class="o">.</span><span class="n">info</span><span 
class="p">(</span><span class="s1">&#39;Sensor check existence of record: 
</span><span class="si">%s</span><span class="s1">&#39;</span><span 
class="p">,</span> <span class="bp">self</span><span class="o">.</span><span 
class="n">keys</span><span class="p">)</span>
+        <span class="n">hook</span> <span class="o">=</span> <span 
class="n">CassandraHook</span><span class="p">(</span><span 
class="bp">self</span><span class="o">.</span><span 
class="n">cassandra_conn_id</span><span class="p">)</span>
+        <span class="k">return</span> <span class="n">hook</span><span 
class="o">.</span><span class="n">record_exists</span><span 
class="p">(</span><span class="bp">self</span><span class="o">.</span><span 
class="n">table</span><span class="p">,</span> <span 
class="bp">self</span><span class="o">.</span><span class="n">keys</span><span 
class="p">)</span></div></div>
+</pre></div>
+
+           </div>
+           
+          </div>
+          <footer>
+  
+
+  <hr/>
+
+  <div role="contentinfo">
+    <p>
+
+    </p>
+  </div>
+  Built with <a href="http://sphinx-doc.org/";>Sphinx</a> using a <a 
href="https://github.com/rtfd/sphinx_rtd_theme";>theme</a> provided by <a 
href="https://readthedocs.org";>Read the Docs</a>. 
+
+</footer>
+
+        </div>
+      </div>
+
+    </section>
+
+  </div>
+  
+
+
+  
+
+    
+    
+      <script type="text/javascript" id="documentation_options" 
data-url_root="../../../../" 
src="../../../../_static/documentation_options.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
+
+  
+
+  <script type="text/javascript" 
src="../../../../_static/js/theme.js"></script>
+
+  <script type="text/javascript">
+      jQuery(function () {
+          SphinxRtdTheme.Navigation.enable(true);
+      });
+  </script> 
+
+</body>
+</html>
\ No newline at end of file

http://git-wip-us.apache.org/repos/asf/incubator-airflow-site/blob/1f06fa0e/_modules/airflow/contrib/sensors/datadog_sensor.html
----------------------------------------------------------------------
diff --git a/_modules/airflow/contrib/sensors/datadog_sensor.html 
b/_modules/airflow/contrib/sensors/datadog_sensor.html
index 31e7fbd..0596ed4 100644
--- a/_modules/airflow/contrib/sensors/datadog_sensor.html
+++ b/_modules/airflow/contrib/sensors/datadog_sensor.html
@@ -272,20 +272,13 @@
 
   
 
-    <script type="text/javascript">
-        var DOCUMENTATION_OPTIONS = {
-            URL_ROOT:'../../../../',
-            VERSION:'',
-            LANGUAGE:'None',
-            COLLAPSE_INDEX:false,
-            FILE_SUFFIX:'.html',
-            HAS_SOURCE:  true,
-            SOURCELINK_SUFFIX: '.txt'
-        };
-    </script>
-      <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
+    
+      <script type="text/javascript" id="documentation_options" 
data-url_root="../../../../" 
src="../../../../_static/documentation_options.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
 
   
 

http://git-wip-us.apache.org/repos/asf/incubator-airflow-site/blob/1f06fa0e/_modules/airflow/contrib/sensors/emr_base_sensor.html
----------------------------------------------------------------------
diff --git a/_modules/airflow/contrib/sensors/emr_base_sensor.html 
b/_modules/airflow/contrib/sensors/emr_base_sensor.html
index 4b2bbaa..0cc4783 100644
--- a/_modules/airflow/contrib/sensors/emr_base_sensor.html
+++ b/_modules/airflow/contrib/sensors/emr_base_sensor.html
@@ -245,20 +245,13 @@
 
   
 
-    <script type="text/javascript">
-        var DOCUMENTATION_OPTIONS = {
-            URL_ROOT:'../../../../',
-            VERSION:'',
-            LANGUAGE:'None',
-            COLLAPSE_INDEX:false,
-            FILE_SUFFIX:'.html',
-            HAS_SOURCE:  true,
-            SOURCELINK_SUFFIX: '.txt'
-        };
-    </script>
-      <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
+    
+      <script type="text/javascript" id="documentation_options" 
data-url_root="../../../../" 
src="../../../../_static/documentation_options.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
 
   
 

http://git-wip-us.apache.org/repos/asf/incubator-airflow-site/blob/1f06fa0e/_modules/airflow/contrib/sensors/emr_job_flow_sensor.html
----------------------------------------------------------------------
diff --git a/_modules/airflow/contrib/sensors/emr_job_flow_sensor.html 
b/_modules/airflow/contrib/sensors/emr_job_flow_sensor.html
index be5d091..f22db8d 100644
--- a/_modules/airflow/contrib/sensors/emr_job_flow_sensor.html
+++ b/_modules/airflow/contrib/sensors/emr_job_flow_sensor.html
@@ -243,20 +243,13 @@
 
   
 
-    <script type="text/javascript">
-        var DOCUMENTATION_OPTIONS = {
-            URL_ROOT:'../../../../',
-            VERSION:'',
-            LANGUAGE:'None',
-            COLLAPSE_INDEX:false,
-            FILE_SUFFIX:'.html',
-            HAS_SOURCE:  true,
-            SOURCELINK_SUFFIX: '.txt'
-        };
-    </script>
-      <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
+    
+      <script type="text/javascript" id="documentation_options" 
data-url_root="../../../../" 
src="../../../../_static/documentation_options.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
 
   
 

http://git-wip-us.apache.org/repos/asf/incubator-airflow-site/blob/1f06fa0e/_modules/airflow/contrib/sensors/emr_step_sensor.html
----------------------------------------------------------------------
diff --git a/_modules/airflow/contrib/sensors/emr_step_sensor.html 
b/_modules/airflow/contrib/sensors/emr_step_sensor.html
index f0bf133..1985c82 100644
--- a/_modules/airflow/contrib/sensors/emr_step_sensor.html
+++ b/_modules/airflow/contrib/sensors/emr_step_sensor.html
@@ -193,8 +193,8 @@
 <span class="sd">    :type step_id: string</span>
 <span class="sd">    &quot;&quot;&quot;</span>
 
-    <span class="n">NON_TERMINAL_STATES</span> <span class="o">=</span> <span 
class="p">[</span><span class="s1">&#39;PENDING&#39;</span><span 
class="p">,</span> <span class="s1">&#39;RUNNING&#39;</span><span 
class="p">,</span> <span class="s1">&#39;CONTINUE&#39;</span><span 
class="p">]</span>
-    <span class="n">FAILED_STATE</span> <span class="o">=</span> <span 
class="p">[</span><span class="s1">&#39;CANCELLED&#39;</span><span 
class="p">,</span> <span class="s1">&#39;FAILED&#39;</span><span 
class="p">]</span>
+    <span class="n">NON_TERMINAL_STATES</span> <span class="o">=</span> <span 
class="p">[</span><span class="s1">&#39;PENDING&#39;</span><span 
class="p">,</span> <span class="s1">&#39;RUNNING&#39;</span><span 
class="p">,</span> <span class="s1">&#39;CONTINUE&#39;</span><span 
class="p">,</span> <span class="s1">&#39;CANCEL_PENDING&#39;</span><span 
class="p">]</span>
+    <span class="n">FAILED_STATE</span> <span class="o">=</span> <span 
class="p">[</span><span class="s1">&#39;CANCELLED&#39;</span><span 
class="p">,</span> <span class="s1">&#39;FAILED&#39;</span><span 
class="p">,</span> <span class="s1">&#39;INTERRUPTED&#39;</span><span 
class="p">]</span>
     <span class="n">template_fields</span> <span class="o">=</span> <span 
class="p">[</span><span class="s1">&#39;job_flow_id&#39;</span><span 
class="p">,</span> <span class="s1">&#39;step_id&#39;</span><span 
class="p">]</span>
     <span class="n">template_ext</span> <span class="o">=</span> <span 
class="p">()</span>
 
@@ -246,20 +246,13 @@
 
   
 
-    <script type="text/javascript">
-        var DOCUMENTATION_OPTIONS = {
-            URL_ROOT:'../../../../',
-            VERSION:'',
-            LANGUAGE:'None',
-            COLLAPSE_INDEX:false,
-            FILE_SUFFIX:'.html',
-            HAS_SOURCE:  true,
-            SOURCELINK_SUFFIX: '.txt'
-        };
-    </script>
-      <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
+    
+      <script type="text/javascript" id="documentation_options" 
data-url_root="../../../../" 
src="../../../../_static/documentation_options.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
 
   
 

http://git-wip-us.apache.org/repos/asf/incubator-airflow-site/blob/1f06fa0e/_modules/airflow/contrib/sensors/file_sensor.html
----------------------------------------------------------------------
diff --git a/_modules/airflow/contrib/sensors/file_sensor.html 
b/_modules/airflow/contrib/sensors/file_sensor.html
index 5985b0c..ef6da69 100644
--- a/_modules/airflow/contrib/sensors/file_sensor.html
+++ b/_modules/airflow/contrib/sensors/file_sensor.html
@@ -260,20 +260,13 @@
 
   
 
-    <script type="text/javascript">
-        var DOCUMENTATION_OPTIONS = {
-            URL_ROOT:'../../../../',
-            VERSION:'',
-            LANGUAGE:'None',
-            COLLAPSE_INDEX:false,
-            FILE_SUFFIX:'.html',
-            HAS_SOURCE:  true,
-            SOURCELINK_SUFFIX: '.txt'
-        };
-    </script>
-      <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
+    
+      <script type="text/javascript" id="documentation_options" 
data-url_root="../../../../" 
src="../../../../_static/documentation_options.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
 
   
 

http://git-wip-us.apache.org/repos/asf/incubator-airflow-site/blob/1f06fa0e/_modules/airflow/contrib/sensors/ftp_sensor.html
----------------------------------------------------------------------
diff --git a/_modules/airflow/contrib/sensors/ftp_sensor.html 
b/_modules/airflow/contrib/sensors/ftp_sensor.html
index 977fb5c..c8609e6 100644
--- a/_modules/airflow/contrib/sensors/ftp_sensor.html
+++ b/_modules/airflow/contrib/sensors/ftp_sensor.html
@@ -256,20 +256,13 @@
 
   
 
-    <script type="text/javascript">
-        var DOCUMENTATION_OPTIONS = {
-            URL_ROOT:'../../../../',
-            VERSION:'',
-            LANGUAGE:'None',
-            COLLAPSE_INDEX:false,
-            FILE_SUFFIX:'.html',
-            HAS_SOURCE:  true,
-            SOURCELINK_SUFFIX: '.txt'
-        };
-    </script>
-      <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
+    
+      <script type="text/javascript" id="documentation_options" 
data-url_root="../../../../" 
src="../../../../_static/documentation_options.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
 
   
 

http://git-wip-us.apache.org/repos/asf/incubator-airflow-site/blob/1f06fa0e/_modules/airflow/contrib/sensors/gcs_sensor.html
----------------------------------------------------------------------
diff --git a/_modules/airflow/contrib/sensors/gcs_sensor.html 
b/_modules/airflow/contrib/sensors/gcs_sensor.html
index 12530ff..44b7234 100644
--- a/_modules/airflow/contrib/sensors/gcs_sensor.html
+++ b/_modules/airflow/contrib/sensors/gcs_sensor.html
@@ -355,20 +355,13 @@
 
   
 
-    <script type="text/javascript">
-        var DOCUMENTATION_OPTIONS = {
-            URL_ROOT:'../../../../',
-            VERSION:'',
-            LANGUAGE:'None',
-            COLLAPSE_INDEX:false,
-            FILE_SUFFIX:'.html',
-            HAS_SOURCE:  true,
-            SOURCELINK_SUFFIX: '.txt'
-        };
-    </script>
-      <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
+    
+      <script type="text/javascript" id="documentation_options" 
data-url_root="../../../../" 
src="../../../../_static/documentation_options.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
 
   
 

http://git-wip-us.apache.org/repos/asf/incubator-airflow-site/blob/1f06fa0e/_modules/airflow/contrib/sensors/hdfs_sensor.html
----------------------------------------------------------------------
diff --git a/_modules/airflow/contrib/sensors/hdfs_sensor.html 
b/_modules/airflow/contrib/sensors/hdfs_sensor.html
index fba42db..5b344b6 100644
--- a/_modules/airflow/contrib/sensors/hdfs_sensor.html
+++ b/_modules/airflow/contrib/sensors/hdfs_sensor.html
@@ -267,20 +267,13 @@
 
   
 
-    <script type="text/javascript">
-        var DOCUMENTATION_OPTIONS = {
-            URL_ROOT:'../../../../',
-            VERSION:'',
-            LANGUAGE:'None',
-            COLLAPSE_INDEX:false,
-            FILE_SUFFIX:'.html',
-            HAS_SOURCE:  true,
-            SOURCELINK_SUFFIX: '.txt'
-        };
-    </script>
-      <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
+    
+      <script type="text/javascript" id="documentation_options" 
data-url_root="../../../../" 
src="../../../../_static/documentation_options.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
 
   
 

http://git-wip-us.apache.org/repos/asf/incubator-airflow-site/blob/1f06fa0e/_modules/airflow/contrib/sensors/jira_sensor.html
----------------------------------------------------------------------
diff --git a/_modules/airflow/contrib/sensors/jira_sensor.html 
b/_modules/airflow/contrib/sensors/jira_sensor.html
index 0130988..9a4f942 100644
--- a/_modules/airflow/contrib/sensors/jira_sensor.html
+++ b/_modules/airflow/contrib/sensors/jira_sensor.html
@@ -340,20 +340,13 @@
 
   
 
-    <script type="text/javascript">
-        var DOCUMENTATION_OPTIONS = {
-            URL_ROOT:'../../../../',
-            VERSION:'',
-            LANGUAGE:'None',
-            COLLAPSE_INDEX:false,
-            FILE_SUFFIX:'.html',
-            HAS_SOURCE:  true,
-            SOURCELINK_SUFFIX: '.txt'
-        };
-    </script>
-      <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
+    
+      <script type="text/javascript" id="documentation_options" 
data-url_root="../../../../" 
src="../../../../_static/documentation_options.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
 
   
 

http://git-wip-us.apache.org/repos/asf/incubator-airflow-site/blob/1f06fa0e/_modules/airflow/contrib/sensors/pubsub_sensor.html
----------------------------------------------------------------------
diff --git a/_modules/airflow/contrib/sensors/pubsub_sensor.html 
b/_modules/airflow/contrib/sensors/pubsub_sensor.html
index 2865450..ab662a2 100644
--- a/_modules/airflow/contrib/sensors/pubsub_sensor.html
+++ b/_modules/airflow/contrib/sensors/pubsub_sensor.html
@@ -294,20 +294,13 @@
 
   
 
-    <script type="text/javascript">
-        var DOCUMENTATION_OPTIONS = {
-            URL_ROOT:'../../../../',
-            VERSION:'',
-            LANGUAGE:'None',
-            COLLAPSE_INDEX:false,
-            FILE_SUFFIX:'.html',
-            HAS_SOURCE:  true,
-            SOURCELINK_SUFFIX: '.txt'
-        };
-    </script>
-      <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
+    
+      <script type="text/javascript" id="documentation_options" 
data-url_root="../../../../" 
src="../../../../_static/documentation_options.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
 
   
 

http://git-wip-us.apache.org/repos/asf/incubator-airflow-site/blob/1f06fa0e/_modules/airflow/contrib/sensors/qubole_sensor.html
----------------------------------------------------------------------
diff --git a/_modules/airflow/contrib/sensors/qubole_sensor.html 
b/_modules/airflow/contrib/sensors/qubole_sensor.html
index 4dd678a..533757f 100644
--- a/_modules/airflow/contrib/sensors/qubole_sensor.html
+++ b/_modules/airflow/contrib/sensors/qubole_sensor.html
@@ -277,20 +277,13 @@
 
   
 
-    <script type="text/javascript">
-        var DOCUMENTATION_OPTIONS = {
-            URL_ROOT:'../../../../',
-            VERSION:'',
-            LANGUAGE:'None',
-            COLLAPSE_INDEX:false,
-            FILE_SUFFIX:'.html',
-            HAS_SOURCE:  true,
-            SOURCELINK_SUFFIX: '.txt'
-        };
-    </script>
-      <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
+    
+      <script type="text/javascript" id="documentation_options" 
data-url_root="../../../../" 
src="../../../../_static/documentation_options.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
 
   
 

http://git-wip-us.apache.org/repos/asf/incubator-airflow-site/blob/1f06fa0e/_modules/airflow/contrib/sensors/redis_key_sensor.html
----------------------------------------------------------------------
diff --git a/_modules/airflow/contrib/sensors/redis_key_sensor.html 
b/_modules/airflow/contrib/sensors/redis_key_sensor.html
index e90f7d5..62d5c49 100644
--- a/_modules/airflow/contrib/sensors/redis_key_sensor.html
+++ b/_modules/airflow/contrib/sensors/redis_key_sensor.html
@@ -236,20 +236,13 @@
 
   
 
-    <script type="text/javascript">
-        var DOCUMENTATION_OPTIONS = {
-            URL_ROOT:'../../../../',
-            VERSION:'',
-            LANGUAGE:'None',
-            COLLAPSE_INDEX:false,
-            FILE_SUFFIX:'.html',
-            HAS_SOURCE:  true,
-            SOURCELINK_SUFFIX: '.txt'
-        };
-    </script>
-      <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
-      <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
+    
+      <script type="text/javascript" id="documentation_options" 
data-url_root="../../../../" 
src="../../../../_static/documentation_options.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
 
   
 

http://git-wip-us.apache.org/repos/asf/incubator-airflow-site/blob/1f06fa0e/_modules/airflow/contrib/sensors/sagemaker_base_sensor.html
----------------------------------------------------------------------
diff --git a/_modules/airflow/contrib/sensors/sagemaker_base_sensor.html 
b/_modules/airflow/contrib/sensors/sagemaker_base_sensor.html
new file mode 100644
index 0000000..023c0a9
--- /dev/null
+++ b/_modules/airflow/contrib/sensors/sagemaker_base_sensor.html
@@ -0,0 +1,285 @@
+
+
+<!DOCTYPE html>
+<!--[if IE 8]><html class="no-js lt-ie9" lang="en" > <![endif]-->
+<!--[if gt IE 8]><!--> <html class="no-js" lang="en" > <!--<![endif]-->
+<head>
+  <meta charset="utf-8">
+  
+  <meta name="viewport" content="width=device-width, initial-scale=1.0">
+  
+  <title>airflow.contrib.sensors.sagemaker_base_sensor &mdash; Airflow 
Documentation</title>
+  
+
+  
+  
+  
+  
+
+  
+
+  
+  
+    
+
+  
+
+  <link rel="stylesheet" href="../../../../_static/css/theme.css" 
type="text/css" />
+  <link rel="stylesheet" href="../../../../_static/pygments.css" 
type="text/css" />
+    <link rel="index" title="Index" href="../../../../genindex.html" />
+    <link rel="search" title="Search" href="../../../../search.html" /> 
+
+  
+  <script src="../../../../_static/js/modernizr.min.js"></script>
+
+</head>
+
+<body class="wy-body-for-nav">
+
+   
+  <div class="wy-grid-for-nav">
+
+    
+    <nav data-toggle="wy-nav-shift" class="wy-nav-side">
+      <div class="wy-side-scroll">
+        <div class="wy-side-nav-search">
+          
+
+          
+            <a href="../../../../index.html" class="icon icon-home"> Airflow
+          
+
+          
+          </a>
+
+          
+            
+            
+          
+
+          
+<div role="search">
+  <form id="rtd-search-form" class="wy-form" action="../../../../search.html" 
method="get">
+    <input type="text" name="q" placeholder="Search docs" />
+    <input type="hidden" name="check_keywords" value="yes" />
+    <input type="hidden" name="area" value="default" />
+  </form>
+</div>
+
+          
+        </div>
+
+        <div class="wy-menu wy-menu-vertical" data-spy="affix" 
role="navigation" aria-label="main navigation">
+          
+            
+            
+              
+            
+            
+              <ul>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../project.html">Project</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../license.html">License</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../start.html">Quick Start</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../installation.html">Installation</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../tutorial.html">Tutorial</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../howto/index.html">How-to Guides</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../ui.html">UI / Screenshots</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../concepts.html">Concepts</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../profiling.html">Data Profiling</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../cli.html">Command Line Interface</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../scheduler.html">Scheduling &amp; Triggers</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../plugins.html">Plugins</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../security.html">Security</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../timezone.html">Time zones</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../api.html">Experimental Rest API</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../integration.html">Integration</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../lineage.html">Lineage</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../faq.html">FAQ</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../code.html">API Reference</a></li>
+</ul>
+
+            
+          
+        </div>
+      </div>
+    </nav>
+
+    <section data-toggle="wy-nav-shift" class="wy-nav-content-wrap">
+
+      
+      <nav class="wy-nav-top" aria-label="top navigation">
+        
+          <i data-toggle="wy-nav-top" class="fa fa-bars"></i>
+          <a href="../../../../index.html">Airflow</a>
+        
+      </nav>
+
+
+      <div class="wy-nav-content">
+        
+        <div class="rst-content">
+        
+          
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+<div role="navigation" aria-label="breadcrumbs navigation">
+
+  <ul class="wy-breadcrumbs">
+    
+      <li><a href="../../../../index.html">Docs</a> &raquo;</li>
+        
+          <li><a href="../../../index.html">Module code</a> &raquo;</li>
+        
+      <li>airflow.contrib.sensors.sagemaker_base_sensor</li>
+    
+    
+      <li class="wy-breadcrumbs-aside">
+        
+      </li>
+    
+  </ul>
+
+  
+  <hr/>
+</div>
+          <div role="main" class="document" itemscope="itemscope" 
itemtype="http://schema.org/Article";>
+           <div itemprop="articleBody">
+            
+  <h1>Source code for airflow.contrib.sensors.sagemaker_base_sensor</h1><div 
class="highlight"><pre>
+<span></span><span class="c1"># -*- coding: utf-8 -*-</span>
+<span class="c1">#</span>
+<span class="c1"># Licensed to the Apache Software Foundation (ASF) under 
one</span>
+<span class="c1"># or more contributor license agreements.  See the NOTICE 
file</span>
+<span class="c1"># distributed with this work for additional information</span>
+<span class="c1"># regarding copyright ownership.  The ASF licenses this 
file</span>
+<span class="c1"># to you under the Apache License, Version 2.0 (the</span>
+<span class="c1"># &quot;License&quot;); you may not use this file except in 
compliance</span>
+<span class="c1"># with the License.  You may obtain a copy of the License 
at</span>
+<span class="c1">#</span>
+<span class="c1">#   http://www.apache.org/licenses/LICENSE-2.0</span>
+<span class="c1">#</span>
+<span class="c1"># Unless required by applicable law or agreed to in 
writing,</span>
+<span class="c1"># software distributed under the License is distributed on 
an</span>
+<span class="c1"># &quot;AS IS&quot; BASIS, WITHOUT WARRANTIES OR CONDITIONS 
OF ANY</span>
+<span class="c1"># KIND, either express or implied.  See the License for 
the</span>
+<span class="c1"># specific language governing permissions and 
limitations</span>
+<span class="c1"># under the License.</span>
+<span class="kn">from</span> <span 
class="nn">airflow.sensors.base_sensor_operator</span> <span 
class="k">import</span> <span class="n">BaseSensorOperator</span>
+<span class="kn">from</span> <span class="nn">airflow.utils.decorators</span> 
<span class="k">import</span> <span class="n">apply_defaults</span>
+<span class="kn">from</span> <span class="nn">airflow.exceptions</span> <span 
class="k">import</span> <span class="n">AirflowException</span>
+
+
+<div class="viewcode-block" id="SageMakerBaseSensor"><a class="viewcode-back" 
href="../../../../code.html#airflow.contrib.sensors.sagemaker_base_sensor.SageMakerBaseSensor">[docs]</a><span
 class="k">class</span> <span class="nc">SageMakerBaseSensor</span><span 
class="p">(</span><span class="n">BaseSensorOperator</span><span 
class="p">):</span>
+    <span class="sd">&quot;&quot;&quot;</span>
+<span class="sd">    Contains general sensor behavior for SageMaker.</span>
+<span class="sd">    Subclasses should implement 
get_sagemaker_response()</span>
+<span class="sd">    and state_from_response() methods.</span>
+<span class="sd">    Subclasses should also implement NON_TERMINAL_STATES and 
FAILED_STATE methods.</span>
+<span class="sd">    &quot;&quot;&quot;</span>
+    <span class="n">ui_color</span> <span class="o">=</span> <span 
class="s1">&#39;#ededed&#39;</span>
+
+    <span class="nd">@apply_defaults</span>
+    <span class="k">def</span> <span class="nf">__init__</span><span 
class="p">(</span>
+            <span class="bp">self</span><span class="p">,</span>
+            <span class="n">aws_conn_id</span><span class="o">=</span><span 
class="s1">&#39;aws_default&#39;</span><span class="p">,</span>
+            <span class="o">*</span><span class="n">args</span><span 
class="p">,</span> <span class="o">**</span><span class="n">kwargs</span><span 
class="p">):</span>
+        <span class="nb">super</span><span class="p">(</span><span 
class="n">SageMakerBaseSensor</span><span class="p">,</span> <span 
class="bp">self</span><span class="p">)</span><span class="o">.</span><span 
class="fm">__init__</span><span class="p">(</span><span class="o">*</span><span 
class="n">args</span><span class="p">,</span> <span class="o">**</span><span 
class="n">kwargs</span><span class="p">)</span>
+        <span class="bp">self</span><span class="o">.</span><span 
class="n">aws_conn_id</span> <span class="o">=</span> <span 
class="n">aws_conn_id</span>
+
+<div class="viewcode-block" id="SageMakerBaseSensor.poke"><a 
class="viewcode-back" 
href="../../../../code.html#airflow.contrib.sensors.sagemaker_base_sensor.SageMakerBaseSensor.poke">[docs]</a>
    <span class="k">def</span> <span class="nf">poke</span><span 
class="p">(</span><span class="bp">self</span><span class="p">,</span> <span 
class="n">context</span><span class="p">):</span>
+        <span class="n">response</span> <span class="o">=</span> <span 
class="bp">self</span><span class="o">.</span><span 
class="n">get_sagemaker_response</span><span class="p">()</span>
+
+        <span class="k">if</span> <span class="ow">not</span> <span 
class="n">response</span><span class="p">[</span><span 
class="s1">&#39;ResponseMetadata&#39;</span><span class="p">][</span><span 
class="s1">&#39;HTTPStatusCode&#39;</span><span class="p">]</span> <span 
class="o">==</span> <span class="mi">200</span><span class="p">:</span>
+            <span class="bp">self</span><span class="o">.</span><span 
class="n">log</span><span class="o">.</span><span class="n">info</span><span 
class="p">(</span><span class="s1">&#39;Bad HTTP response: </span><span 
class="si">%s</span><span class="s1">&#39;</span><span class="p">,</span> <span 
class="n">response</span><span class="p">)</span>
+            <span class="k">return</span> <span class="kc">False</span>
+
+        <span class="n">state</span> <span class="o">=</span> <span 
class="bp">self</span><span class="o">.</span><span 
class="n">state_from_response</span><span class="p">(</span><span 
class="n">response</span><span class="p">)</span>
+
+        <span class="bp">self</span><span class="o">.</span><span 
class="n">log</span><span class="o">.</span><span class="n">info</span><span 
class="p">(</span><span class="s1">&#39;Job currently </span><span 
class="si">%s</span><span class="s1">&#39;</span><span class="p">,</span> <span 
class="n">state</span><span class="p">)</span>
+
+        <span class="k">if</span> <span class="n">state</span> <span 
class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span 
class="n">non_terminal_states</span><span class="p">():</span>
+            <span class="k">return</span> <span class="kc">False</span>
+
+        <span class="k">if</span> <span class="n">state</span> <span 
class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span 
class="n">failed_states</span><span class="p">():</span>
+            <span class="n">failed_reason</span> <span class="o">=</span> 
<span class="bp">self</span><span class="o">.</span><span 
class="n">get_failed_reason_from_response</span><span class="p">(</span><span 
class="n">response</span><span class="p">)</span>
+            <span class="k">raise</span> <span 
class="n">AirflowException</span><span class="p">(</span><span 
class="s1">&#39;Sagemaker job failed for the following reason: </span><span 
class="si">%s</span><span class="s1">&#39;</span>
+                                   <span class="o">%</span> <span 
class="n">failed_reason</span><span class="p">)</span>
+        <span class="k">return</span> <span class="kc">True</span></div>
+
+    <span class="k">def</span> <span 
class="nf">non_terminal_states</span><span class="p">(</span><span 
class="bp">self</span><span class="p">):</span>
+        <span class="k">raise</span> <span 
class="ne">NotImplementedError</span><span class="p">(</span><span 
class="s1">&#39;Please implement non_terminal_states() in 
subclass&#39;</span><span class="p">)</span>
+
+    <span class="k">def</span> <span class="nf">failed_states</span><span 
class="p">(</span><span class="bp">self</span><span class="p">):</span>
+        <span class="k">raise</span> <span 
class="ne">NotImplementedError</span><span class="p">(</span><span 
class="s1">&#39;Please implement failed_states() in subclass&#39;</span><span 
class="p">)</span>
+
+    <span class="k">def</span> <span 
class="nf">get_sagemaker_response</span><span class="p">(</span><span 
class="bp">self</span><span class="p">):</span>
+        <span class="k">raise</span> <span 
class="ne">NotImplementedError</span><span class="p">(</span><span 
class="s1">&#39;Please implement get_sagemaker_response() in 
subclass&#39;</span><span class="p">)</span>
+
+    <span class="k">def</span> <span 
class="nf">get_failed_reason_from_response</span><span class="p">(</span><span 
class="bp">self</span><span class="p">,</span> <span 
class="n">response</span><span class="p">):</span>
+        <span class="k">return</span> <span class="s1">&#39;Unknown&#39;</span>
+
+    <span class="k">def</span> <span 
class="nf">state_from_response</span><span class="p">(</span><span 
class="bp">self</span><span class="p">,</span> <span 
class="n">response</span><span class="p">):</span>
+        <span class="k">raise</span> <span 
class="ne">NotImplementedError</span><span class="p">(</span><span 
class="s1">&#39;Please implement state_from_response() in 
subclass&#39;</span><span class="p">)</span></div>
+</pre></div>
+
+           </div>
+           
+          </div>
+          <footer>
+  
+
+  <hr/>
+
+  <div role="contentinfo">
+    <p>
+
+    </p>
+  </div>
+  Built with <a href="http://sphinx-doc.org/";>Sphinx</a> using a <a 
href="https://github.com/rtfd/sphinx_rtd_theme";>theme</a> provided by <a 
href="https://readthedocs.org";>Read the Docs</a>. 
+
+</footer>
+
+        </div>
+      </div>
+
+    </section>
+
+  </div>
+  
+
+
+  
+
+    
+    
+      <script type="text/javascript" id="documentation_options" 
data-url_root="../../../../" 
src="../../../../_static/documentation_options.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
+
+  
+
+  <script type="text/javascript" 
src="../../../../_static/js/theme.js"></script>
+
+  <script type="text/javascript">
+      jQuery(function () {
+          SphinxRtdTheme.Navigation.enable(true);
+      });
+  </script> 
+
+</body>
+</html>
\ No newline at end of file

http://git-wip-us.apache.org/repos/asf/incubator-airflow-site/blob/1f06fa0e/_modules/airflow/contrib/sensors/sagemaker_endpoint_sensor.html
----------------------------------------------------------------------
diff --git a/_modules/airflow/contrib/sensors/sagemaker_endpoint_sensor.html 
b/_modules/airflow/contrib/sensors/sagemaker_endpoint_sensor.html
new file mode 100644
index 0000000..68a1c19
--- /dev/null
+++ b/_modules/airflow/contrib/sensors/sagemaker_endpoint_sensor.html
@@ -0,0 +1,272 @@
+
+
+<!DOCTYPE html>
+<!--[if IE 8]><html class="no-js lt-ie9" lang="en" > <![endif]-->
+<!--[if gt IE 8]><!--> <html class="no-js" lang="en" > <!--<![endif]-->
+<head>
+  <meta charset="utf-8">
+  
+  <meta name="viewport" content="width=device-width, initial-scale=1.0">
+  
+  <title>airflow.contrib.sensors.sagemaker_endpoint_sensor &mdash; Airflow 
Documentation</title>
+  
+
+  
+  
+  
+  
+
+  
+
+  
+  
+    
+
+  
+
+  <link rel="stylesheet" href="../../../../_static/css/theme.css" 
type="text/css" />
+  <link rel="stylesheet" href="../../../../_static/pygments.css" 
type="text/css" />
+    <link rel="index" title="Index" href="../../../../genindex.html" />
+    <link rel="search" title="Search" href="../../../../search.html" /> 
+
+  
+  <script src="../../../../_static/js/modernizr.min.js"></script>
+
+</head>
+
+<body class="wy-body-for-nav">
+
+   
+  <div class="wy-grid-for-nav">
+
+    
+    <nav data-toggle="wy-nav-shift" class="wy-nav-side">
+      <div class="wy-side-scroll">
+        <div class="wy-side-nav-search">
+          
+
+          
+            <a href="../../../../index.html" class="icon icon-home"> Airflow
+          
+
+          
+          </a>
+
+          
+            
+            
+          
+
+          
+<div role="search">
+  <form id="rtd-search-form" class="wy-form" action="../../../../search.html" 
method="get">
+    <input type="text" name="q" placeholder="Search docs" />
+    <input type="hidden" name="check_keywords" value="yes" />
+    <input type="hidden" name="area" value="default" />
+  </form>
+</div>
+
+          
+        </div>
+
+        <div class="wy-menu wy-menu-vertical" data-spy="affix" 
role="navigation" aria-label="main navigation">
+          
+            
+            
+              
+            
+            
+              <ul>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../project.html">Project</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../license.html">License</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../start.html">Quick Start</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../installation.html">Installation</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../tutorial.html">Tutorial</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../howto/index.html">How-to Guides</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../ui.html">UI / Screenshots</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../concepts.html">Concepts</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../profiling.html">Data Profiling</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../cli.html">Command Line Interface</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../scheduler.html">Scheduling &amp; Triggers</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../plugins.html">Plugins</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../security.html">Security</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../timezone.html">Time zones</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../api.html">Experimental Rest API</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../integration.html">Integration</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../lineage.html">Lineage</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../faq.html">FAQ</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../code.html">API Reference</a></li>
+</ul>
+
+            
+          
+        </div>
+      </div>
+    </nav>
+
+    <section data-toggle="wy-nav-shift" class="wy-nav-content-wrap">
+
+      
+      <nav class="wy-nav-top" aria-label="top navigation">
+        
+          <i data-toggle="wy-nav-top" class="fa fa-bars"></i>
+          <a href="../../../../index.html">Airflow</a>
+        
+      </nav>
+
+
+      <div class="wy-nav-content">
+        
+        <div class="rst-content">
+        
+          
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+<div role="navigation" aria-label="breadcrumbs navigation">
+
+  <ul class="wy-breadcrumbs">
+    
+      <li><a href="../../../../index.html">Docs</a> &raquo;</li>
+        
+          <li><a href="../../../index.html">Module code</a> &raquo;</li>
+        
+      <li>airflow.contrib.sensors.sagemaker_endpoint_sensor</li>
+    
+    
+      <li class="wy-breadcrumbs-aside">
+        
+      </li>
+    
+  </ul>
+
+  
+  <hr/>
+</div>
+          <div role="main" class="document" itemscope="itemscope" 
itemtype="http://schema.org/Article";>
+           <div itemprop="articleBody">
+            
+  <h1>Source code for 
airflow.contrib.sensors.sagemaker_endpoint_sensor</h1><div 
class="highlight"><pre>
+<span></span><span class="c1"># -*- coding: utf-8 -*-</span>
+<span class="c1">#</span>
+<span class="c1"># Licensed to the Apache Software Foundation (ASF) under 
one</span>
+<span class="c1"># or more contributor license agreements.  See the NOTICE 
file</span>
+<span class="c1"># distributed with this work for additional information</span>
+<span class="c1"># regarding copyright ownership.  The ASF licenses this 
file</span>
+<span class="c1"># to you under the Apache License, Version 2.0 (the</span>
+<span class="c1"># &quot;License&quot;); you may not use this file except in 
compliance</span>
+<span class="c1"># with the License.  You may obtain a copy of the License 
at</span>
+<span class="c1">#</span>
+<span class="c1">#   http://www.apache.org/licenses/LICENSE-2.0</span>
+<span class="c1">#</span>
+<span class="c1"># Unless required by applicable law or agreed to in 
writing,</span>
+<span class="c1"># software distributed under the License is distributed on 
an</span>
+<span class="c1"># &quot;AS IS&quot; BASIS, WITHOUT WARRANTIES OR CONDITIONS 
OF ANY</span>
+<span class="c1"># KIND, either express or implied.  See the License for 
the</span>
+<span class="c1"># specific language governing permissions and 
limitations</span>
+<span class="c1"># under the License.</span>
+
+<span class="kn">from</span> <span 
class="nn">airflow.contrib.hooks.sagemaker_hook</span> <span 
class="k">import</span> <span class="n">SageMakerHook</span>
+<span class="kn">from</span> <span 
class="nn">airflow.contrib.sensors.sagemaker_base_sensor</span> <span 
class="k">import</span> <span class="n">SageMakerBaseSensor</span>
+<span class="kn">from</span> <span class="nn">airflow.utils.decorators</span> 
<span class="k">import</span> <span class="n">apply_defaults</span>
+
+
+<div class="viewcode-block" id="SageMakerEndpointSensor"><a 
class="viewcode-back" 
href="../../../../code.html#airflow.contrib.sensors.sagemaker_endpoint_sensor.SageMakerEndpointSensor">[docs]</a><span
 class="k">class</span> <span class="nc">SageMakerEndpointSensor</span><span 
class="p">(</span><span class="n">SageMakerBaseSensor</span><span 
class="p">):</span>
+    <span class="sd">&quot;&quot;&quot;</span>
+<span class="sd">    Asks for the state of the endpoint state until it reaches 
a terminal state.</span>
+<span class="sd">    If it fails the sensor errors, the task fails.</span>
+
+<span class="sd">    :param job_name: job_name of the endpoint instance to 
check the state of</span>
+<span class="sd">    :type job_name: str</span>
+<span class="sd">    &quot;&quot;&quot;</span>
+
+    <span class="n">template_fields</span> <span class="o">=</span> <span 
class="p">[</span><span class="s1">&#39;endpoint_name&#39;</span><span 
class="p">]</span>
+    <span class="n">template_ext</span> <span class="o">=</span> <span 
class="p">()</span>
+
+    <span class="nd">@apply_defaults</span>
+    <span class="k">def</span> <span class="nf">__init__</span><span 
class="p">(</span><span class="bp">self</span><span class="p">,</span>
+                 <span class="n">endpoint_name</span><span class="p">,</span>
+                 <span class="o">*</span><span class="n">args</span><span 
class="p">,</span>
+                 <span class="o">**</span><span class="n">kwargs</span><span 
class="p">):</span>
+        <span class="nb">super</span><span class="p">(</span><span 
class="n">SageMakerEndpointSensor</span><span class="p">,</span> <span 
class="bp">self</span><span class="p">)</span><span class="o">.</span><span 
class="fm">__init__</span><span class="p">(</span><span class="o">*</span><span 
class="n">args</span><span class="p">,</span> <span class="o">**</span><span 
class="n">kwargs</span><span class="p">)</span>
+        <span class="bp">self</span><span class="o">.</span><span 
class="n">endpoint_name</span> <span class="o">=</span> <span 
class="n">endpoint_name</span>
+
+    <span class="k">def</span> <span 
class="nf">non_terminal_states</span><span class="p">(</span><span 
class="bp">self</span><span class="p">):</span>
+        <span class="k">return</span> <span 
class="n">SageMakerHook</span><span class="o">.</span><span 
class="n">endpoint_non_terminal_states</span>
+
+    <span class="k">def</span> <span class="nf">failed_states</span><span 
class="p">(</span><span class="bp">self</span><span class="p">):</span>
+        <span class="k">return</span> <span 
class="n">SageMakerHook</span><span class="o">.</span><span 
class="n">failed_states</span>
+
+    <span class="k">def</span> <span 
class="nf">get_sagemaker_response</span><span class="p">(</span><span 
class="bp">self</span><span class="p">):</span>
+        <span class="n">sagemaker</span> <span class="o">=</span> <span 
class="n">SageMakerHook</span><span class="p">(</span><span 
class="n">aws_conn_id</span><span class="o">=</span><span 
class="bp">self</span><span class="o">.</span><span 
class="n">aws_conn_id</span><span class="p">)</span>
+
+        <span class="bp">self</span><span class="o">.</span><span 
class="n">log</span><span class="o">.</span><span class="n">info</span><span 
class="p">(</span><span class="s1">&#39;Poking Sagemaker Endpoint </span><span 
class="si">%s</span><span class="s1">&#39;</span><span class="p">,</span> <span 
class="bp">self</span><span class="o">.</span><span 
class="n">endpoint_name</span><span class="p">)</span>
+        <span class="k">return</span> <span class="n">sagemaker</span><span 
class="o">.</span><span class="n">describe_endpoint</span><span 
class="p">(</span><span class="bp">self</span><span class="o">.</span><span 
class="n">endpoint_name</span><span class="p">)</span>
+
+    <span class="k">def</span> <span 
class="nf">get_failed_reason_from_response</span><span class="p">(</span><span 
class="bp">self</span><span class="p">,</span> <span 
class="n">response</span><span class="p">):</span>
+        <span class="k">return</span> <span class="n">response</span><span 
class="p">[</span><span class="s1">&#39;FailureReason&#39;</span><span 
class="p">]</span>
+
+    <span class="k">def</span> <span 
class="nf">state_from_response</span><span class="p">(</span><span 
class="bp">self</span><span class="p">,</span> <span 
class="n">response</span><span class="p">):</span>
+        <span class="k">return</span> <span class="n">response</span><span 
class="p">[</span><span class="s1">&#39;EndpointStatus&#39;</span><span 
class="p">]</span></div>
+</pre></div>
+
+           </div>
+           
+          </div>
+          <footer>
+  
+
+  <hr/>
+
+  <div role="contentinfo">
+    <p>
+
+    </p>
+  </div>
+  Built with <a href="http://sphinx-doc.org/";>Sphinx</a> using a <a 
href="https://github.com/rtfd/sphinx_rtd_theme";>theme</a> provided by <a 
href="https://readthedocs.org";>Read the Docs</a>. 
+
+</footer>
+
+        </div>
+      </div>
+
+    </section>
+
+  </div>
+  
+
+
+  
+
+    
+    
+      <script type="text/javascript" id="documentation_options" 
data-url_root="../../../../" 
src="../../../../_static/documentation_options.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
+        <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+    
+
+  
+
+  <script type="text/javascript" 
src="../../../../_static/js/theme.js"></script>
+
+  <script type="text/javascript">
+      jQuery(function () {
+          SphinxRtdTheme.Navigation.enable(true);
+      });
+  </script> 
+
+</body>
+</html>
\ No newline at end of file

Reply via email to