Fix RBAC for subpages in response models

src/zenml/zen_server/rbac/utils.py

@@ -494,8 +494,12 @@ def get_subresources_for_model(
     # We previously used `dict(model)` here, but that lead to issues with
     # models overwriting `__getattr__`, this `model.__iter__()` has the same
     # results though.
-    for _, value in model.__iter__():
-        resources.update(_get_subresources_for_value(value))
+    if isinstance(model, Page):
+        for item in model:
+            resources.update(_get_subresources_for_value(item))
+    else:
+        for _, value in model.__iter__():
+            resources.update(_get_subresources_for_value(value))
 
     return resources
 

tests/integration/examples/discord/steps/data_loader.py

@@ -13,10 +13,11 @@
 #  permissions and limitations under the License.
 
 from typing import Tuple
-from typing_extensions import Annotated
+
 import numpy as np
 from sklearn.datasets import load_digits
 from sklearn.model_selection import train_test_split
+from typing_extensions import Annotated
 
 from zenml import step
 
@@ -27,7 +28,7 @@ def digits_data_loader() -> (
         Annotated[np.ndarray, "X_train"],
         Annotated[np.ndarray, "X_test"],
         Annotated[np.ndarray, "y_train"],
-        Annotated[np.ndarray, "y_test"]
+        Annotated[np.ndarray, "y_test"],
     ]
 ):
     """Loads the digits dataset as a tuple of flattened numpy arrays."""