laipeng/reflex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix pyright

Browse Source
This commit is contained in:
Elijah 2024-10-04 11:05:21 +00:00
parent 8e0044d856
commit d1cd68d16d
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
reflex/components/datadisplay/shiki_code_block.py
View File

@ -15,13 +15,14 @@ from reflex.components.lucide.icon import Icon
from reflex.components.radix.themes.layout.box import Box from reflex.components.radix.themes.layout.box import Box
from reflex.event import call_script from reflex.event import call_script
from reflex.style import Style from reflex.style import Style
from reflex.utils.exceptions import VarTypeError
from reflex.utils.imports import ImportVar from reflex.utils.imports import ImportVar
from reflex.vars.base import LiteralVar, Var from reflex.vars.base import LiteralVar, Var
from reflex.vars.function import FunctionStringVar from reflex.vars.function import FunctionStringVar
from reflex.vars.sequence import StringVar, string_replace_operation from reflex.vars.sequence import StringVar, string_replace_operation
def copy_script(id: str, code: str) -> Any: def copy_script(id: str, code: StringVar | str) -> Any:
"""Copy script for the code block. """Copy script for the code block.
Args: Args:
@ -773,10 +774,10 @@ class ShikiHighLevelCodeBlock(ShikiCodeBlock):
return language return language
@staticmethod @staticmethod
def _strip_transformer_triggers(code: str | Var) -> StringVar | str: def _strip_transformer_triggers(code: str | StringVar) -> StringVar | str:
if not isinstance(code, (Var, str)): if not isinstance(code, (StringVar, str)):
raise ValueError( raise VarTypeError(
f"code should be string literal or a Var type. Got {type(code)} instead." f"code should be string literal or a StringVar type. Got {type(code)} instead."
) )
if isinstance(code, Var): if isinstance(code, Var):