Skip to content

Commit b377231

Browse files
fix: add new backend tools to basestruct test (#6925)
## 📝 Summary <!-- Provide a concise summary of what this pull request is addressing. If this PR fixes any issues, list them here by number (e.g., Fixes #123). --> This adds new tests backend tools to the basestruct test which checks if any variables in the args or output are using BaseStruct instead of msgspec.Struct (which is not pydantic compatible and will throw a type error) ## 🔍 Description of Changes <!-- Detail the specific changes made in this pull request. Explain the problem addressed and how it was resolved. If applicable, provide before and after comparisons, screenshots, or any relevant details to help reviewers understand the changes easily. --> ## 📋 Checklist - [x] I have read the [contributor guidelines](https://github.com/marimo-team/marimo/blob/main/CONTRIBUTING.md). - [ ] For large changes, or changes that affect the public API: this change was discussed or approved through an issue, on [Discord](https://marimo.io/discord?ref=pr), or the community [discussions](https://github.com/marimo-team/marimo/discussions) (Please provide a link if applicable). - [ ] I have added tests for the changes made. - [x] I have run the code and verified that it works as expected. --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1 parent 4c8d0b6 commit b377231

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

tests/_utils/test_msgspec_basestruct.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,8 @@
33
import msgspec
44

55
from marimo._ai._tools.tools.cells import (
6+
GetCellOutputArgs,
7+
GetCellOutputOutput,
68
GetCellRuntimeDataArgs,
79
GetCellRuntimeDataOutput,
810
GetLightweightCellMapArgs,
@@ -16,6 +18,10 @@
1618
GetNotebookErrorsArgs,
1719
GetNotebookErrorsOutput,
1820
)
21+
from marimo._ai._tools.tools.lint import (
22+
LintNotebookArgs,
23+
LintNotebookOutput,
24+
)
1925
from marimo._ai._tools.tools.notebooks import (
2026
GetActiveNotebooksOutput,
2127
)
@@ -36,6 +42,10 @@
3642
GetNotebookErrorsArgs,
3743
GetNotebookErrorsOutput,
3844
GetActiveNotebooksOutput,
45+
LintNotebookArgs,
46+
LintNotebookOutput,
47+
GetCellOutputArgs,
48+
GetCellOutputOutput,
3949
]
4050

4151

0 commit comments

Comments
 (0)