minor import fixes

This commit is contained in:
Hardik Shah 2024-08-26 14:21:35 -07:00
parent dc433f6c90
commit c3708859aa
7 changed files with 16 additions and 11 deletions

View file

@ -8,7 +8,7 @@ from codeshield.cs import CodeShield
from termcolor import cprint
from .base import ShieldResponse, TextShield
from llama_toolchain.safety.api.datatypes import * # noqa: F403
from llama_toolchain.safety.api import * # noqa: F403
class CodeScannerShield(TextShield):