Fix flake8 failure

saimidu · saimidu · commit 56be3c2552b3 · 2021-10-22T10:37:10.000-07:00
diff --git a/test/unit/test_default_inference_handler.py b/test/unit/test_default_inference_handler.py
@@ -102,7 +102,7 @@ def test_default_model_fn_no_model_file(inference_handler, listdir_return_value)
             with pytest.raises(
                 ValueError, match=r"Exactly one .pth or .pt file is required for PyTorch models: .*"
             ):
-                model = inference_handler.default_model_fn("model_dir")
+                inference_handler.default_model_fn("model_dir")
 
 
 def test_default_input_fn_json(inference_handler, tensor):
