diff --git a/sdk/python/tests/integration/feature_repos/repo_configuration.py b/sdk/python/tests/integration/feature_repos/repo_configuration.py index e34ece8474..9a0784b3af 100644 --- a/sdk/python/tests/integration/feature_repos/repo_configuration.py +++ b/sdk/python/tests/integration/feature_repos/repo_configuration.py @@ -488,36 +488,36 @@ def setup(self): Permission( name="offline_permissions_perm", types=Permission, - policy=RoleBasedPolicy(roles=["reader"]), - actions=[AuthzedAction.QUERY_OFFLINE], + policy=RoleBasedPolicy(roles=["reader", "writer"]), + actions=[AuthzedAction.QUERY_OFFLINE, AuthzedAction.WRITE_OFFLINE], ), Permission( name="offline_entities_perm", types=Entity, with_subclasses=False, - policy=RoleBasedPolicy(roles=["reader"]), - actions=[AuthzedAction.QUERY_OFFLINE], + policy=RoleBasedPolicy(roles=["reader", "writer"]), + actions=[AuthzedAction.QUERY_OFFLINE, AuthzedAction.WRITE_OFFLINE], ), Permission( name="offline_fv_perm", types=FeatureView, with_subclasses=False, - policy=RoleBasedPolicy(roles=["reader"]), - actions=[AuthzedAction.QUERY_OFFLINE], + policy=RoleBasedPolicy(roles=["reader", "writer"]), + actions=[AuthzedAction.QUERY_OFFLINE, AuthzedAction.WRITE_OFFLINE], ), Permission( name="offline_odfv_perm", types=OnDemandFeatureView, with_subclasses=False, - policy=RoleBasedPolicy(roles=["reader"]), - actions=[AuthzedAction.QUERY_OFFLINE], + policy=RoleBasedPolicy(roles=["reader", "writer"]), + actions=[AuthzedAction.QUERY_OFFLINE, AuthzedAction.WRITE_OFFLINE], ), Permission( name="offline_sfv_perm", types=StreamFeatureView, with_subclasses=False, - policy=RoleBasedPolicy(roles=["reader"]), - actions=[AuthzedAction.QUERY_OFFLINE], + policy=RoleBasedPolicy(roles=["reader", "writer"]), + actions=[AuthzedAction.QUERY_OFFLINE, AuthzedAction.WRITE_OFFLINE], ), ] self.feature_store.apply(permissions_list)