Type check collection.py

ThePalaceProject · Oct 25, 2023 · 9a6da1d · 9a6da1d
1 parent 349f690
commit 9a6da1d
Show file tree

Hide file tree

Showing 12 changed files with 175 additions and 109 deletions.
diff --git a/core/metadata_layer.py b/core/metadata_layer.py
@@ -554,7 +554,7 @@ def add_to_pool(self, db: Session, pool: LicensePool):
 
 
 class TimestampData:
-    CLEAR_VALUE = Timestamp.CLEAR_VALUE
+    CLEAR_VALUE = Timestamp.CLEAR_VALUE  # type: ignore[has-type]
 
     def __init__(
         self, start=None, finish=None, achievements=None, counter=None, exception=None

diff --git a/core/model/classification.py b/core/model/classification.py
@@ -56,16 +56,16 @@ class Subject(Base):
     OVERDRIVE = Classifier.OVERDRIVE  # Overdrive's classification system
     BISAC = Classifier.BISAC
     BIC = Classifier.BIC  # BIC Subject Categories
-    TAG = Classifier.TAG  # Folksonomic tags.
-    FREEFORM_AUDIENCE = Classifier.FREEFORM_AUDIENCE
+    TAG: str = Classifier.TAG  # Folksonomic tags.
+    FREEFORM_AUDIENCE: str = Classifier.FREEFORM_AUDIENCE
     NYPL_APPEAL = Classifier.NYPL_APPEAL
 
     # Types with terms that are suitable for search.
     TYPES_FOR_SEARCH = [FAST, OVERDRIVE, BISAC, TAG]
 
     AXIS_360_AUDIENCE = Classifier.AXIS_360_AUDIENCE
     GRADE_LEVEL = Classifier.GRADE_LEVEL
-    AGE_RANGE = Classifier.AGE_RANGE
+    AGE_RANGE: str = Classifier.AGE_RANGE
     LEXILE_SCORE = Classifier.LEXILE_SCORE
     ATOS_SCORE = Classifier.ATOS_SCORE
     INTEREST_LEVEL = Classifier.INTEREST_LEVEL