minor

Signed-off-by: Panwar, Ashish <[email protected]>
microsoft · Nov 7, 2023 · 1381736 · 1381736
1 parent 371d697
commit 1381736
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 8 deletions.
diff --git a/.pylintrc b/.pylintrc
@@ -170,7 +170,8 @@ disable=abstract-method,
         access-member-before-definition,
         use-dict-literal,
         consider-using-generator,
-        unnecessary-dunder-call
+        unnecessary-dunder-call,
+	import-outside-toplevel
 
 
 

diff --git a/simulator/events/batch_stage_end_event.py b/simulator/events/batch_stage_end_event.py
@@ -8,12 +8,6 @@
 from simulator.scheduler import BaseGlobalScheduler
 from simulator.types import EventType
 
-from simulator.events.batch_end_event import BatchEndEvent
-from simulator.events.batch_stage_arrival_event import BatchStageArrivalEvent
-from simulator.events.replica_stage_schedule_event import (
-    ReplicaStageScheduleEvent,
-)
-
 logger = logging.getLogger(__name__)
 
 
@@ -43,7 +37,11 @@ def event_type(self):
     def handle_event(
         self, scheduler: BaseGlobalScheduler, metrics_store: MetricsStore
     ) -> List[BaseEvent]:
-
+        from simulator.events.batch_end_event import BatchEndEvent
+        from simulator.events.batch_stage_arrival_event import BatchStageArrivalEvent
+        from simulator.events.replica_stage_schedule_event import (
+            ReplicaStageScheduleEvent,
+        )
         scheduler.get_replica_stage_scheduler(
             self._replica_id, self._stage_id
         ).on_stage_end()