Merge pull request #52 from chriscardillo/v0.17.0

v0.17.0
pipeline-tools · Feb 7, 2023 · 043a567 · 043a567
2 parents 88304ef + 44fb8db
commit 043a567
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 10 deletions.
diff --git a/README.md b/README.md
@@ -149,7 +149,6 @@ While `METADATA.yml` will always be the primary source of truth for a DAG or Tas
 Here's an example using `create_dag`, where instead of metadata we use `create_dag` arguments:
 
 ```py
-import airflow
 from datetime import timedelta
 from airflow.utils.dates import days_ago
 from gusty import create_dag
@@ -192,7 +191,6 @@ If you have multiple gusty DAGs located inside of a single directory, you can co
 Let's adjust the above `create_dag` example to use `create_dags` instead:
 
 ```py
-import airflow
 from datetime import timedelta
 from airflow.utils.dates import days_ago
 from gusty import create_dags

diff --git a/gusty/__init__.py b/gusty/__init__.py
@@ -1,4 +1,3 @@
-import airflow  # noqa
 from gusty.building import GustyBuilder
 
 

diff --git a/gusty/building.py b/gusty/building.py
@@ -1,5 +1,5 @@
-import os, inspect, airflow
-from airflow import DAG
+import os, inspect
+from airflow.models import BaseOperator, DAG
 from absql import Runner
 from functools import partial
 from gusty.errors import NonexistentDagDirError
@@ -150,7 +150,7 @@ def build_task(spec, level_id, schematic):
         k: v
         for k, v in spec.items()
         if k
-        in inspect.signature(airflow.models.BaseOperator.__init__).parameters.keys()
+        in inspect.signature(BaseOperator.__init__).parameters.keys()
         or k in _get_operator_parameters(operator)
         or k in _get_operator_parameters(operator.__base__)
     }
@@ -289,7 +289,7 @@ def __init__(self, dag_dir, **kwargs):
                 if k in AVAILABLE_WAIT_FOR_PARAMS
                 or k
                 in inspect.signature(
-                    airflow.models.BaseOperator.__init__
+                    BaseOperator.__init__
                 ).parameters.keys()
             }
             self.wait_for_defaults.update(user_wait_for_defaults)

diff --git a/gusty/importing.py b/gusty/importing.py
@@ -1,11 +1,12 @@
-import os, sys, pkgutil, itertools, airflow, importlib
+import os, sys, pkgutil, itertools, importlib
+from airflow.version import version as _airflow_version
 from inflection import underscore
 
 ############
 ## Params ##
 ############
 
-airflow_version = int(str(airflow.__version__).split(".")[0])
+airflow_version = int(_airflow_version.split(".")[0])
 
 ###########################
 ## Operator Import Logic ##

diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="gusty",
-    version="0.16.0",
+    version="0.17.0",
     author="Chris Cardillo, Michael Chow, David Robinson",
     author_email="cfcardillo23@gmail.com",
     description="Making DAG construction easier",
Original file line number	Diff line number	Diff line change
		@@ -1,4 +1,3 @@
		import airflow # noqa
		from gusty.building import GustyBuilder


Expand Down