From a353524d4f83ebfb0648c06a9074b8b0a5cf7b24 Mon Sep 17 00:00:00 2001 From: Stefan Nica Date: Wed, 19 Feb 2025 13:41:24 +0100 Subject: [PATCH] Fix some linter errors --- src/zenml/zen_server/rbac/utils.py | 4 ++-- src/zenml/zen_stores/sql_zen_store.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/zenml/zen_server/rbac/utils.py b/src/zenml/zen_server/rbac/utils.py index eeec59725d..fc128fa706 100644 --- a/src/zenml/zen_server/rbac/utils.py +++ b/src/zenml/zen_server/rbac/utils.py @@ -178,7 +178,7 @@ def _dehydrate_value( return value -def has_permissions_for_model(model: AnyResponse, action: Action) -> bool: +def has_permissions_for_model(model: AnyModel, action: Action) -> bool: """If the active user has permissions to perform the action on the model. Args: @@ -342,7 +342,7 @@ def get_allowed_resource_ids( if not server_config().rbac_enabled: return None - if ResourceType(resource_type).is_workspace_scoped and not workspace_id: + if ResourceType(resource_type).is_workspace_scoped() and not workspace_id: raise ValueError( "Workspace ID is required to list workspace scoped resources." ) diff --git a/src/zenml/zen_stores/sql_zen_store.py b/src/zenml/zen_stores/sql_zen_store.py index a97ab6787f..e55d1cc561 100644 --- a/src/zenml/zen_stores/sql_zen_store.py +++ b/src/zenml/zen_stores/sql_zen_store.py @@ -11352,14 +11352,14 @@ def _attach_tags_to_resource( try: tag = self.get_tag(tag_name) self._verify_workspace( - expected_workspace_id=resource.workspace_id, # type: ignore[attr-defined] + expected_workspace_id=resource.workspace_id, obj=tag, ) except KeyError: tag = self.create_tag( TagRequest( name=tag_name, - workspace=resource.workspace_id, # type: ignore[attr-defined] + workspace=resource.workspace_id, ) ) try: