final changes before commit

godatadriven · Aug 25, 2024 · 4a7ca09 · 4a7ca09
1 parent 1949a31
commit 4a7ca09
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/chapter08/dags/01_python.py b/chapter08/dags/01_python.py
@@ -11,7 +11,7 @@
 
 MOVIELENS_HOST = os.environ.get("MOVIELENS_HOST", "movielens")
 MOVIELENS_SCHEMA = os.environ.get("MOVIELENS_SCHEMA", "http")
-MOVIELENS_PORT = os.environ.get("MOVIELENS_PORT", "5000")
+MOVIELENS_PORT = os.environ.get("MOVIELENS_PORT", "8081")
 
 MOVIELENS_USER = os.environ["MOVIELENS_USER"]
 MOVIELENS_PASSWORD = os.environ["MOVIELENS_PASSWORD"]

diff --git a/chapter08/dags/05_deferrable_sensor.py b/chapter08/dags/05_deferrable_sensor.py
@@ -1,7 +1,7 @@
 from datetime import datetime, timedelta
 from airflow import DAG
 
-from custom.operators import MovielensFetchOperator
+from custom.operators import MovielensFetchRatingsOperator
 from custom.triggers import MovielensSensorAsync
 
 with DAG(

diff --git a/chapter08/dags/custom/triggers.py b/chapter08/dags/custom/triggers.py
@@ -35,7 +35,7 @@ def __init__(self,
     def execute(self, context: Context) -> None:
 
         self.defer(
-            trigger=MovielensRatingsTrigger(
+            trigger=MovielensTrigger(
                 conn_id=self._conn_id,
                 sleep_interval=self._sleep_interval,
                 start_date=self._start_date,
@@ -67,7 +67,7 @@ def __init__(self,
 
 
     def serialize(self):
-        return ("custom.triggers.MovielensRatingsTrigger", {
+        return ("custom.triggers.MovielensTrigger", {
                 "sleep_interval": self._sleep_interval,
                 "conn_id": self._conn_id,
                 "start_date": self._start_date,