[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
unitaryai · Jan 16, 2025 · 1a2b2e9 · 1a2b2e9
1 parent 4659611
commit 1a2b2e9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/run_prediction.py b/run_prediction.py
@@ -33,7 +33,8 @@ def run(model_name, input_obj, dest_file, from_ckpt, device="cpu"):
     res = model.predict(text)
 
     res_df = pd.DataFrame(
-        res, index=[text] if isinstance(text, str) else text  # pyright: ignore[reportArgumentType]
+        res,
+        index=[text] if isinstance(text, str) else text,  # pyright: ignore[reportArgumentType]
     ).round(5)
     print(res_df)
     if dest_file is not None: