From ab8a220faa7dd70fa9f0ec98b8568e1c5662fe0e Mon Sep 17 00:00:00 2001 From: Kate Plawiak Date: Mon, 22 Jul 2024 22:03:05 -0700 Subject: [PATCH] add missing license part --- llama_toolchain/safety/shields/llama_guard.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/llama_toolchain/safety/shields/llama_guard.py b/llama_toolchain/safety/shields/llama_guard.py index 12aadc305..d6154f027 100644 --- a/llama_toolchain/safety/shields/llama_guard.py +++ b/llama_toolchain/safety/shields/llama_guard.py @@ -1,3 +1,4 @@ +# Copyright (c) Meta Platforms, Inc. and affiliates. # All rights reserved. # # This source code is licensed under the terms described in the LICENSE file in @@ -217,7 +218,7 @@ class LlamaGuardShield(ShieldBase): raise ValueError(f"Unexpected response: {response}") async def run(self, messages: List[Message]) -> ShieldResponse: - if self.disable_input_check and messages[-1].role == Role.user.value: + if self.disable_input_check and messages[-1].role == Role.user.value: return ShieldResponse( shield_type=BuiltinShield.llama_guard, is_violation=False )