fixed import issues

MasoniteFramework · Nov 29, 2019 · 271d9ff · 271d9ff
1 parent 4d9afb8
commit 271d9ff
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 10 deletions.
diff --git a/src/scheduler/commands/ScheduleRunCommand.py b/src/scheduler/commands/ScheduleRunCommand.py
@@ -1,7 +1,8 @@
 """ A ScheduleRunCommand Command """
-from cleo import Command
-from scheduler.Task import Task
 import pendulum
+from cleo import Command
+
+from ..Task import Task
 
 
 class ScheduleRunCommand(Command):
@@ -15,7 +16,7 @@ class ScheduleRunCommand(Command):
     def handle(self):
         from wsgi import container as app
         tasks = app.collect(Task)
-        
+
         for task_key, task_class in tasks.items():
             # Resolve the task with the container
             if self.option('task') != 'None':

diff --git a/tests/test_basic.py b/tests/test_basic.py
diff --git a/tests/test_scheduler.py b/tests/test_scheduler.py
@@ -1,6 +1,6 @@
 import pytest
 import pendulum
-from scheduler.Task import Task
+from src.scheduler.Task import Task
 
 class MockTask(Task):
     run_every = '5 minutes'