Add contributing guideline and mypy config (#122)

2023-05-23 17:58:51 -07:00
parent 3f942acfe1
commit a283ec2eec
16 changed files with 128 additions and 44 deletions
--- a/cacheflow/core/scheduler.py
+++ b/cacheflow/core/scheduler.py
@@ -87,7 +87,7 @@ class Scheduler:
    def has_unfinished_seqs(self) -> bool:
        return self.waiting or self.running or self.swapped

-    def _schedule(self) -> Tuple[SchedulerOutputs, List[int]]:
+    def _schedule(self) -> Tuple[SchedulerOutputs, List[str]]:
        # Blocks that need to be swaped or copied before model execution.
        blocks_to_swap_in: Dict[int, int] = {}
        blocks_to_swap_out: Dict[int, int] = {}