Skip to content

Commit 541e4cf

Browse files
committed
Fix lint error
1 parent b164662 commit 541e4cf

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/trustyai/language/detoxify/tmarco.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,7 @@ def load_models(self, experts: list[str] = None, expert_weights: list = None):
111111
expert_models = []
112112
for expert in experts:
113113
# Load TMaRCO models
114-
if (expert == "trustyai/gplus" or expert == "trustyai/gminus"):
114+
if expert in ["trustyai/gplus", "trustyai/gminus"]:
115115
expert = BartForConditionalGeneration.from_pretrained(
116116
expert,
117117
forced_bos_token_id=self.tokenizer.bos_token_id,

0 commit comments

Comments
 (0)