Fix imports

AudiusProject · Nov 7, 2023 · 5bfd49b · 5bfd49b
1 parent 519456c
commit 5bfd49b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/packages/discovery-provider/src/premium_content/signature.py b/packages/discovery-provider/src/premium_content/signature.py
@@ -1,6 +1,8 @@
 import json
 from datetime import datetime
-from typing import NotRequired, Optional, TypedDict
+from typing import Optional, TypedDict
+
+from typing_extensions import NotRequired
 
 from src.api_helpers import generate_signature
 from src.premium_content.premium_content_types import PremiumContentType

diff --git a/packages/discovery-provider/src/tasks/entity_manager/entities/track.py b/packages/discovery-provider/src/tasks/entity_manager/entities/track.py
@@ -16,7 +16,7 @@
 from src.models.users.user import User
 from src.premium_content.premium_content_access_checker import (
     PremiumContentAccessBatchArgs,
-    PremiumContentAccessChecker,
+    premium_content_access_checker,
 )
 from src.premium_content.premium_content_constants import USDC_PURCHASE_KEY
 from src.tasks.entity_manager.utils import (
@@ -478,7 +478,7 @@ def validate_remixability(params: ManageEntityParameters):
         )
     )
     premium_content_batch_access = (
-        PremiumContentAccessChecker.check_access_for_batch(session, args)
+        premium_content_access_checker.check_access_for_batch(session, args)
     )
     if "track" not in premium_content_batch_access:
         return
@@ -494,7 +494,7 @@ def validate_remixability(params: ManageEntityParameters):
 
 
 def get_remix_parent_track_ids(track_metadata):
-    if not "remix_of" in track_metadata:
+    if "remix_of" not in track_metadata:
         return
     if not isinstance(track_metadata["remix_of"], dict):
         return