fixed minor comment

rudderlabs · Oct 9, 2024 · ca4812c · ca4812c
1 parent 3ec2ae2
commit ca4812c
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -2,3 +2,4 @@ apache-airflow == 2.10.0
 requests == 2.32.3
 setuptools == 65.5.1 # not directly required, pinned by Snyk to avoid a vulnerability
 pytest==7.3.1
+ruff==0.6.8
diff --git a/rudder_airflow_provider/hooks/rudderstack.py b/rudder_airflow_provider/hooks/rudderstack.py
@@ -116,7 +116,7 @@ def make_request(
                 response.raise_for_status()
                 return response.json()
             except requests.RequestException as e:
-                logging.error(f"Request to url: {url} failed: {e}")
+                self.log.error(f"Request to url: {url} failed: {e}")
                 if num_retries == self.request_max_retries:
                     break
                 num_retries += 1

diff --git a/rudder_airflow_provider/operators/rudderstack.py b/rudder_airflow_provider/operators/rudderstack.py
@@ -113,7 +113,7 @@ def execute(self, context):
         )
         profile_run_id = rs_profiles_hook.start_profile_run(self.profile_id)
         if self.wait_for_completion:
-            logging.info(
+            self.log.info(
                 f"Poll and wait for profiles run to finish for profilesId: {self.profile_id}, runId: {profile_run_id}"
             )
             rs_profiles_hook.poll_profile_run(self.profile_id, profile_run_id)