Merge pull request #1094 from airbnb/pig_operator

Adding to inits
apache · Feb 29, 2016 · 936f4ae · 936f4ae
2 parents f6eebb0 + 1c5aa66
commit 936f4ae
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/airflow/hooks/__init__.py b/airflow/hooks/__init__.py
@@ -11,6 +11,7 @@
     ],
     'hdfs_hook': ['HDFSHook'],
     'webhdfs_hook': ['WebHDFSHook'],
+    'pig_hook': ['PigCliHook'],
     'mysql_hook': ['MySqlHook'],
     'postgres_hook': ['PostgresHook'],
     'presto_hook': ['PrestoHook'],

diff --git a/airflow/hooks/pig_hook.py b/airflow/hooks/pig_hook.py
@@ -72,5 +72,5 @@ def run_cli(self, pig, verbose=True):
     def kill(self):
         if hasattr(self, 'sp'):
             if self.sp.poll() is None:
-                print("Killing the Hive job")
+                print("Killing the Pig job")
                 self.sp.kill()
diff --git a/airflow/operators/__init__.py b/airflow/operators/__init__.py
@@ -19,6 +19,7 @@
         'ShortCircuitOperator',
     ],
     'hive_operator': ['HiveOperator'],
+    'pig_operator': ['PigOperator'],
     'presto_check_operator': [
         'PrestoCheckOperator',
         'PrestoValueCheckOperator',