nits

Author: swaroopAkkineniWorkos

src/workos/fga.py

@@ -53,9 +53,11 @@ class WarrantQueryListResource(
     warnings: Optional[Sequence[FGAWarning]] = None
 
 
-# Deprecated: Use the Authorization module instead.
-# See: workos.authorization
 class FGAModule(Protocol):
+    """
+    .. deprecated::
+        Use :class:`workos.authorization.AuthorizationModule` instead.
+    """
     def get_resource(
         self, *, resource_type: str, resource_id: str
     ) -> AuthorizationResource:

src/workos/types/authorization/__init__.py

@@ -1,4 +1,5 @@
 from workos.types.authorization.access_check_response import AccessCheckResponse
+from workos.types.authorization.assignment import Assignment
 from workos.types.authorization.environment_role import (
     EnvironmentRole,
     EnvironmentRoleList,