Main

.github/scripts/download_pretrained.py

@@ -52,7 +52,7 @@ def get_model_name_and_weights_from_config(
             if model_name not in model_class_dict:
                 raise KeyError(
                     f"'{model_name}' not a valid model name. Choose from "
-                    f"{str(list(model_class_dict.keys()))} or create"
+                    f"{list(model_class_dict.keys())!s} or create"
                     f"a new class inheriting from this class to support your model."
                 )
 

.github/tests/test_download_pretrained.py

@@ -23,7 +23,9 @@ def test_download_pretrained_lmf_exists_with_model_name():
     config = yaml.load(CONFIG_FPATH)
 
     steps = config.get("pipeline", [])
-    step = list(filter(lambda x: x["name"] == download_pretrained.COMP_NAME, steps))[0]
+    step = list(  # noqa: RUF015
+        filter(lambda x: x["name"] == download_pretrained.COMP_NAME, steps)
+    )[0]
     step["model_name"] = "roberta"
     step["cache_dir"] = "/this/dir"
 
@@ -41,7 +43,9 @@ def test_download_pretrained_unknown_model_name():
     config = yaml.load(CONFIG_FPATH)
 
     steps = config.get("pipeline", [])
-    step = list(filter(lambda x: x["name"] == download_pretrained.COMP_NAME, steps))[0]
+    step = list(  # noqa: RUF015
+        filter(lambda x: x["name"] == download_pretrained.COMP_NAME, steps)
+    )[0]
     step["model_name"] = "unknown"
 
     with tempfile.NamedTemporaryFile("w+") as fp:
@@ -56,7 +60,9 @@ def test_download_pretrained_multiple_model_names():
     config = yaml.load(CONFIG_FPATH)
 
     steps = config.get("pipeline", [])
-    step = list(filter(lambda x: x["name"] == download_pretrained.COMP_NAME, steps))[0]
+    step = list(  # noqa: RUF015
+        filter(lambda x: x["name"] == download_pretrained.COMP_NAME, steps)
+    )[0]
     step_new = deepcopy(step)
     step_new["model_name"] = "roberta"
     steps.append(step_new)
@@ -74,7 +80,9 @@ def test_download_pretrained_with_model_name_and_nondefault_weight():
     config = yaml.load(CONFIG_FPATH)
 
     steps = config.get("pipeline", [])
-    step = list(filter(lambda x: x["name"] == download_pretrained.COMP_NAME, steps))[0]
+    step = list(  # noqa: RUF015
+        filter(lambda x: x["name"] == download_pretrained.COMP_NAME, steps)
+    )[0]
     step["model_name"] = "bert"
     step["model_weights"] = "bert-base-uncased"
 
@@ -91,7 +99,9 @@ def test_download_pretrained_lmf_doesnt_exists():
     config = yaml.load(CONFIG_FPATH)
 
     steps = config.get("pipeline", [])
-    step = list(filter(lambda x: x["name"] == download_pretrained.COMP_NAME, steps))[0]
+    step = list(  # noqa: RUF015
+        filter(lambda x: x["name"] == download_pretrained.COMP_NAME, steps)
+    )[0]
     steps.remove(step)
 
     with tempfile.NamedTemporaryFile("w+") as fp: