Fix DAG import error message (#35)

Signed-off-by: Fredrik Bakken <[email protected]>
treeverse · Aug 15, 2023 · fd52426 · fd52426
1 parent 4f3e790
commit fd52426
Show file tree

Hide file tree

Showing 4 changed files with 0 additions and 4 deletions.
diff --git a/lakefs_provider/operators/commit_operator.py b/lakefs_provider/operators/commit_operator.py
@@ -44,7 +44,6 @@ def __init__(self, lakefs_conn_id: str, repo: str, branch: str, msg: str, metada
         self.branch = branch
         self.msg = msg
         self.metadata = metadata
-        self.task_id = kwargs.get("task_id")
 
     def execute(self, context: Dict[str, Any]) -> Any:
         hook = LakeFSHook(lakefs_conn_id=self.lakefs_conn_id)

diff --git a/lakefs_provider/operators/get_commit_operator.py b/lakefs_provider/operators/get_commit_operator.py
@@ -31,7 +31,6 @@ def __init__(self, lakefs_conn_id: str, repo: str, ref: str, **kwargs: Any) -> N
         self.lakefs_conn_id = lakefs_conn_id
         self.repo = repo
         self.ref = ref
-        self.task_id = kwargs.get("task_id")
 
     def execute(self, context: Dict[str, Any]) -> Any:
         hook = LakeFSHook(lakefs_conn_id=self.lakefs_conn_id)

diff --git a/lakefs_provider/operators/merge_operator.py b/lakefs_provider/operators/merge_operator.py
@@ -48,7 +48,6 @@ def __init__(self, lakefs_conn_id: str, repo: str, source_ref: str, destination_
         self.destination_branch = destination_branch
         self.msg = msg
         self.metadata = metadata
-        self.task_id = kwargs.get("task_id")
 
     def execute(self, context: Dict[str, Any]) -> Any:
         hook = LakeFSHook(lakefs_conn_id=self.lakefs_conn_id)

diff --git a/lakefs_provider/operators/upload_operator.py b/lakefs_provider/operators/upload_operator.py
@@ -39,7 +39,6 @@ def __init__(self, lakefs_conn_id: str, repo: str, branch: str, path: str, conte
         self.branch = branch
         self.path = path
         self.content = content
-        self.task_id = kwargs.get("task_id")
 
     def execute(self, context: Dict[str, Any]) -> Any:
         hook = LakeFSHook(lakefs_conn_id=self.lakefs_conn_id)