|
8 | 8 | import sqlalchemy as sa |
9 | 9 | from aiopg.sa.connection import SAConnection |
10 | 10 | from aiopg.sa.result import RowProxy |
11 | | -from models_library.projects import NodesDict, ProjectID, ProjectType |
| 11 | +from models_library.projects import ProjectID, ProjectType |
12 | 12 | from models_library.projects_nodes import Node |
13 | 13 | from models_library.projects_nodes_io import NodeIDStr |
14 | 14 | from models_library.utils.change_case import camel_to_snake, snake_to_camel |
@@ -195,17 +195,17 @@ async def _upsert_tags_in_project( |
195 | 195 | async def _get_workbench( |
196 | 196 | connection: SAConnection, |
197 | 197 | project_uuid: str, |
198 | | - ) -> NodesDict: |
| 198 | + ) -> dict[str, Any]: |
199 | 199 | project_nodes_repo = ProjectNodesRepo(project_uuid=ProjectID(project_uuid)) |
200 | 200 | exclude_fields = {"node_id", "required_resources", "created", "modified"} |
201 | | - workbench: NodesDict = {} |
| 201 | + workbench: dict[str, Any] = {} |
202 | 202 |
|
203 | 203 | project_nodes = await project_nodes_repo.list(connection) # type: ignore |
204 | 204 | for project_node in project_nodes: |
205 | 205 | node_data = project_node.model_dump( |
206 | 206 | exclude=exclude_fields, exclude_none=True, exclude_unset=True |
207 | 207 | ) |
208 | | - workbench[f"{project_node.node_id}"] = Node.model_validate(node_data) |
| 208 | + workbench[f"{project_node.node_id}"] = node_data |
209 | 209 | return workbench |
210 | 210 |
|
211 | 211 | async def _get_project( |
|
0 commit comments