Workaround Pylint error

ruivieira · ruivieira · commit 5d49d747b6aa · 2024-09-07T20:45:47.000+01:00
diff --git a/src/trustyai/language/detoxify/tmarco.py b/src/trustyai/language/detoxify/tmarco.py
@@ -104,7 +104,7 @@ def __init__(
                     "cuda" if torch.cuda.is_available() else "cpu"
                 )
 
-        def load_models(self, experts: list[str] = None, expert_weights: list = None):
+        def load_models(self, experts: list[str] = None, expert_weights: list = None): # pylint: disable=unsubscriptable-object
             """Load expert models."""
             if expert_weights is not None:
                 self.expert_weights = expert_weights

Original file line number	Diff line number	Diff line change
`@@ -104,7 +104,7 @@ def __init__(`
`104`	`104`	`"cuda" if torch.cuda.is_available() else "cpu"`
`105`	`105`	`)`
`106`	`106`
`107`		`- def load_models(self, experts: list[str] = None, expert_weights: list = None):`
	`107`	`+ def load_models(self, experts: list[str] = None, expert_weights: list = None): # pylint: disable=unsubscriptable-object`
`108`	`108`	`"""Load expert models."""`
`109`	`109`	`if expert_weights is not None:`
`110`	`110`	`self.expert_weights = expert_weights`