dev/readme_snippets/raw/features_demo.py:1:1: INP001 File `dev/readme_snippets/raw/features_demo.py` is part of an implicit namespace package. Add an `__init__.py`.

dev/readme_snippets/raw/features_demo.py:1:1: D100 Missing docstring in public module

dev/scripts/format_snippet_prints.py:1:1: INP001 File `dev/scripts/format_snippet_prints.py` is part of an implicit namespace package. Add an `__init__.py`.

dev/scripts/format_snippet_prints.py:37:9: S102 Use of `exec` detected

src/kajihs_utils/experimental/jax.py:10:5: ANN201 Missing return type annotation for public function `tree_stack`

src/kajihs_utils/experimental/jax.py:10:16: ANN001 Missing type annotation for function argument `trees`

src/kajihs_utils/experimental/jax.py:10:23: ANN001 Missing type annotation for function argument `axis`

src/kajihs_utils/experimental/jax.py:15:5: ANN201 Missing return type annotation for public function `tree_unstack`

src/kajihs_utils/experimental/jax.py:15:5: D103 Missing docstring in public function

src/kajihs_utils/experimental/jax.py:15:18: ANN001 Missing type annotation for function argument `tree`

Type "dict[Unknown, Unknown]" is not assignable to return type "T@make_yamlizable" (reportReturnType)

Type "list[Unknown]" is not assignable to return type "T@make_yamlizable"   Type "list[Unknown]" is not assignable to type "T@make_yamlizable" (reportReturnType)

Type "tuple[Unknown, ...]" is not assignable to return type "T@make_yamlizable"   Type "tuple[Unknown, ...]" is not assignable to type "T@make_yamlizable" (reportReturnType)

Type "Any | str" is not assignable to return type "T@make_yamlizable"   Type "Any | str" is not assignable to type "T@make_yamlizable" (reportReturnType)

Import from `jax` is implicitly relative and will not work if this file is imported as a module   Use a relative import from `.jax` instead   or specify the full module path: `kajihs_utils.experimental.jax` (reportImplicitRelativeImport)

Import "jax.numpy" could not be resolved (reportMissingImports)

Import "jax._src.typing" could not be resolved (reportMissingImports)

Import "jax.typing" could not be resolved (reportMissingImports)

"tree_map" is not a known attribute of module "jax" (reportAttributeAccessIssue)

"tree_flatten" is not a known attribute of module "jax" (reportAttributeAccessIssue)

Process completed with exit code 1.

Return type is unknown (reportUnknownParameterType)

Type of parameter "trees" is unknown (reportUnknownParameterType)

Type annotation is missing for parameter "trees" (reportMissingParameterType)

Type annotation is missing for parameter "axis" (reportMissingParameterType)

Type of parameter "v" is partially unknown (reportUnknownLambdaType)

Return type of lambda is unknown (reportUnknownLambdaType)

Return type, "list[Unknown]", is partially unknown (reportUnknownParameterType)

Type of parameter "tree" is unknown (reportUnknownParameterType)

Type annotation is missing for parameter "tree" (reportMissingParameterType)