Pre-commit

maro/common.py

@@ -1,6 +1,7 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
 
+
 class BaseDecisionEvent:
     """Base class for all decision events.
 

maro/event_buffer/event_buffer.py

@@ -6,12 +6,12 @@
 from collections import defaultdict
 from typing import Callable, List, Optional, cast
 
+from ..common import BaseAction, BaseDecisionEvent
 from .event import ActualEvent, AtomEvent, CascadeEvent
 from .event_linked_list import EventLinkedList
 from .event_pool import EventPool
 from .event_state import EventState
 from .maro_events import MaroEvents
-from ..common import BaseAction, BaseDecisionEvent
 
 
 class EventRecorder:

maro/simulator/core.py

@@ -8,10 +8,10 @@
 from maro.backends.frame import FrameBase, SnapshotList
 from maro.data_lib.dump_csv_converter import DumpConverter
 from maro.event_buffer import ActualEvent, CascadeEvent, EventBuffer, EventState
-from ..common import BaseAction, BaseDecisionEvent
 from maro.streamit import streamit
 from maro.utils.exception.simulator_exception import BusinessEngineNotFoundError
 
+from ..common import BaseAction, BaseDecisionEvent
 from .abs_core import AbsEnv, DecisionMode
 from .scenarios.abs_business_engine import AbsBusinessEngine
 from .utils.common import tick_to_frame_index

maro/simulator/scenarios/cim/business_engine.py

@@ -11,7 +11,7 @@
 
 from maro.backends.frame import FrameBase, SnapshotList
 from maro.data_lib.cim import CimDataContainerWrapper, Order, Stop
-from maro.event_buffer import ActualEvent, AtomEvent, CascadeEvent, EventBuffer, MaroEvents
+from maro.event_buffer import AtomEvent, CascadeEvent, EventBuffer, MaroEvents
 from maro.simulator.scenarios import AbsBusinessEngine
 from maro.simulator.scenarios.helpers import DocableDict
 from maro.simulator.scenarios.matrix_accessor import MatrixAttributeAccessor

maro/simulator/scenarios/citi_bike/business_engine.py

@@ -9,9 +9,9 @@
 import numpy as np
 from dateutil.relativedelta import relativedelta
 from dateutil.tz import gettz
-from maro.backends.frame import FrameBase, SnapshotList
 from yaml import safe_load
 
+from maro.backends.frame import FrameBase, SnapshotList
 from maro.cli.data_pipeline.citi_bike import CitiBikeProcess
 from maro.cli.data_pipeline.utils import chagne_file_path
 from maro.data_lib import BinaryReader
@@ -21,6 +21,7 @@
 from maro.simulator.scenarios.matrix_accessor import MatrixAttributeAccessor
 from maro.utils.exception.cli_exception import CommandError
 from maro.utils.logger import CliLogger
+
 from .adj_loader import load_adj_from_csv
 from .common import Action, BikeReturnPayload, BikeTransferPayload, DecisionEvent
 from .decision_strategy import BikeDecisionStrategy

tests/test_event_buffer.py

@@ -6,16 +6,8 @@
 import unittest
 from typing import Optional
 
-from maro.event_buffer import (
-    ActualEvent,
-    AtomEvent,
-    CascadeEvent,
-    DummyEvent,
-    EventBuffer,
-    EventState,
-    MaroEvents,
-)
 from maro.common import BaseDecisionEvent
+from maro.event_buffer import ActualEvent, AtomEvent, CascadeEvent, DummyEvent, EventBuffer, EventState, MaroEvents
 from maro.event_buffer.event_linked_list import EventLinkedList
 from maro.event_buffer.event_pool import EventPool