kerrychu · kerrychu · Jan 24, 2024 · Jan 24, 2024
diff --git a/requirements.txt b/requirements.txt
diff --git a/src/hooks/__init__.py → schedule/__init__.py b/src/hooks/__init__.py → schedule/__init__.py
diff --git a/src/available_nodes.py → schedule/available_nodes.py b/src/available_nodes.py → schedule/available_nodes.py
@@ -1,11 +1,11 @@
 import os
 from dotenv import load_dotenv
-from src.utils.subprocess_operations import (
+from schedule.utils.subprocess_operations import (
     get_piped_stdout,
     stdout_to_records,
     job_records_to_slack_message,
 )
-from src.hooks.slack import send_slack_message
+from schedule.hooks.slack import send_slack_message
 
 load_dotenv()
 SLACK_WEBHOOK = os.getenv("SLACK_GPU_JOBS_WEBHOOK")

diff --git a/src/gpu_jobs.py → schedule/gpu_jobs.py b/src/gpu_jobs.py → schedule/gpu_jobs.py
@@ -1,11 +1,11 @@
 import os
 from dotenv import load_dotenv
-from src.utils.subprocess_operations import (
+from schedule.utils.subprocess_operations import (
     get_piped_stdout,
     stdout_to_records,
     job_records_to_slack_message,
 )
-from src.hooks.slack import send_slack_message
+from schedule.hooks.slack import send_slack_message
 
 load_dotenv()
 SLACK_WEBHOOK = os.getenv("SLACK_GPU_JOBS_WEBHOOK")

diff --git a/src/utils/__init__.py → schedule/hooks/__init__.py b/src/utils/__init__.py → schedule/hooks/__init__.py
diff --git a/src/hooks/slack.py → schedule/hooks/slack.py b/src/hooks/slack.py → schedule/hooks/slack.py
diff --git a/src/my_jobs.py → schedule/my_jobs.py b/src/my_jobs.py → schedule/my_jobs.py
@@ -2,12 +2,12 @@
 from typing import Optional
 
 from dotenv import load_dotenv
-from src.hooks.slack import send_slack_message
-from src.utils.data_serialization import (
+from schedule.hooks.slack import send_slack_message
+from schedule.utils.data_serialization import (
     read_json_as_job_records,
     write_job_records_to_json,
 )
-from src.utils.subprocess_operations import (
+from schedule.utils.subprocess_operations import (
     stdout_to_job_records,
     RECORDS,
     job_records_to_slack_message,

diff --git a/src/quota.py → schedule/quota.py b/src/quota.py → schedule/quota.py
@@ -1,7 +1,7 @@
 import os
 from dotenv import load_dotenv
-from src.utils.subprocess_operations import get_piped_stdout, stdout_to_quota_records
-from src.hooks.slack import send_slack_message
+from schedule.utils.subprocess_operations import get_piped_stdout, stdout_to_quota_records
+from schedule.hooks.slack import send_slack_message
 
 load_dotenv()
 

diff --git a/tests/test_hooks/__init__.py → schedule/utils/__init__.py b/tests/test_hooks/__init__.py → schedule/utils/__init__.py
diff --git a/src/utils/data_serialization.py → schedule/utils/data_serialization.py b/src/utils/data_serialization.py → schedule/utils/data_serialization.py
diff --git a/src/utils/subprocess_operations.py → schedule/utils/subprocess_operations.py b/src/utils/subprocess_operations.py → schedule/utils/subprocess_operations.py
diff --git a/tests/test_utils/__init__.py → tests/test_schedule/__init__.py b/tests/test_utils/__init__.py → tests/test_schedule/__init__.py
diff --git a/tests/test_gpu_jobs.py → tests/test_schedule/test_gpu_jobs.py b/tests/test_gpu_jobs.py → tests/test_schedule/test_gpu_jobs.py
diff --git a/tests/test_my_jobs.py → tests/test_schedule/test_hooks/__init__.py b/tests/test_my_jobs.py → tests/test_schedule/test_hooks/__init__.py
diff --git a/tests/test_quota.py → tests/test_schedule/test_my_jobs.py b/tests/test_quota.py → tests/test_schedule/test_my_jobs.py
diff --git a/tests/test_schedule/test_quota.py b/tests/test_schedule/test_quota.py
diff --git a/tests/test_schedule/test_utils/__init__.py b/tests/test_schedule/test_utils/__init__.py
diff --git a/.../test_utils/test_subprocess_operations.py → .../test_utils/test_subprocess_operations.py b/.../test_utils/test_subprocess_operations.py → .../test_utils/test_subprocess_operations.py
@@ -1,4 +1,4 @@
-from src.utils.subprocess_operations import (
+from schedule import (
     strip_empty_string,
     job_records_to_slack_message,
 )