langfingaz/anki
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add v3 scheduler to col.sched type union

Browse Source 
Will allow us to catch issues like the custom study one in the future
This commit is contained in:
Damien Elmes 2021-05-19 16:06:52 +10:00
parent 63f1a363af
commit cbd2314e27
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pylib/anki/collection.py
View File

@ -47,7 +47,7 @@ from anki.models import ModelManager, Notetype, NotetypeDict, NotetypeId
from anki.notes import Note, NoteId from anki.notes import Note, NoteId
from anki.scheduler.v1 import Scheduler as V1Scheduler from anki.scheduler.v1 import Scheduler as V1Scheduler
from anki.scheduler.v2 import Scheduler as V2Scheduler from anki.scheduler.v2 import Scheduler as V2Scheduler
from anki.scheduler.v3 import Scheduler as V3TestScheduler from anki.scheduler.v3 import Scheduler as V3Scheduler
from anki.sync import SyncAuth, SyncOutput, SyncStatus from anki.sync import SyncAuth, SyncOutput, SyncStatus
from anki.tags import TagManager from anki.tags import TagManager
from anki.types import assert_exhaustive from anki.types import assert_exhaustive
@ -81,7 +81,7 @@ LegacyUndoResult = Union[None, LegacyCheckpoint, LegacyReviewUndo]
class Collection: class Collection:
sched: Union[V1Scheduler, V2Scheduler] sched: Union[V1Scheduler, V2Scheduler, V3Scheduler]
def __init__( def __init__(
self, self,
@ -155,7 +155,7 @@ class Collection:
self.sched = V1Scheduler(self) self.sched = V1Scheduler(self)
elif ver == 2: elif ver == 2:
if self.is_2021_test_scheduler_enabled(): if self.is_2021_test_scheduler_enabled():
self.sched = V3TestScheduler(self) # type: ignore self.sched = V3Scheduler(self)
else: else:
self.sched = V2Scheduler(self) self.sched = V2Scheduler(self)