renaming

montara-io · Feb 14, 2023 · 8f13869 · 8f13869
1 parent fe24cc9
commit 8f13869
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/common/models.py → common/db_models.py b/common/models.py → common/db_models.py
diff --git a/spark_endpoint/app.py b/spark_endpoint/app.py
@@ -7,7 +7,7 @@
 
 from common.config import app_config
 from common.logger import get_logger
-from common.models import RawEvent, db_session
+from common.db_models import RawEvent, db_session
 
 logger = get_logger()
 

diff --git a/spark_job_processor/app.py b/spark_job_processor/app.py
@@ -6,7 +6,7 @@
 
 from common.config import app_config
 from common.logger import get_logger
-from common.models import RawEvent, SparkJobRun, db_session
+from common.db_models import RawEvent, SparkJobRun, db_session
 from spark_job_processor.events_processor import EventsProcessor
 
 TOPIC_NAME = "JOB_RUN_EVENT"

diff --git a/spark_job_processor/events_processor.py b/spark_job_processor/events_processor.py
@@ -38,7 +38,7 @@ def process_events(
                 4,
             )
 
-        job_run = {
+        processed_data = {
             "id": job_run_id,
             "job_id": job_id,
             "pipeline_id": pipeline_id,
@@ -57,4 +57,4 @@ def process_events(
             "start_time": application.start_time,
             "end_time": application.end_time,
         }
-        return job_run
+        return processed_data
diff --git a/test/test_spark_endpoint.py b/test/test_spark_endpoint.py
@@ -4,7 +4,7 @@
 
 from spark_endpoint.app import app, parse_events
 from common.config import app_config
-from common.models import RawEvent
+from common.db_models import RawEvent
 
 
 class TestWriteEvents(TestCase):