fixed docs and removed unnecessary code

Nike-Inc · Aug 4, 2023 · 62dee19 · 62dee19
1 parent 763a86c
commit 62dee19
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/brickflow/__init__.py b/brickflow/__init__.py
@@ -125,8 +125,10 @@ def setup_logger(name=None):
 
 
 _ilog = setup_logger("brickflow-internal")
-_ilog.setLevel(logging.ERROR)
 """logger for internal logging please do not use outside of brickflow internal"""
+
+_ilog.setLevel(logging.ERROR)
+
 log = setup_logger()
 
 from brickflow.context import ctx

diff --git a/brickflow/codegen/databricks_bundle.py b/brickflow/codegen/databricks_bundle.py
@@ -216,9 +216,7 @@ def supported_types(self) -> List[SupportedResolverTypes]:
 
     def _resolve(self, ref: ResourceReference) -> List[ImportBlock]:
         blocks = []
-        jobs = [job for job in self.databricks_client.jobs.list(name=ref.name)]
-        if len(jobs) == 0:
-            return None
+        jobs = list(self.databricks_client.jobs.list(name=ref.name))
 
         for job in jobs:
             if job.settings is None or job.settings.tags is None: