Formatting

clients/python/tests/test_types.py

@@ -31,8 +31,7 @@ def test_parameters_validation():
 
     # Test temperature
     Parameters(temperature=1)
-    with pytest.raises(ValidationError):
-        Parameters(temperature=0)
+    Parameters(temperature=0)
     with pytest.raises(ValidationError):
         Parameters(temperature=-1)
 

server/lorax_server/utils/layers.py

@@ -231,9 +231,7 @@ def __init__(
         index=None,
     ):
         super().__init__()
-        assert (
-            not memory_efficient_backward
-        ), "memory_efficient_backward is no longer required and the argument is deprecated in 0.37.0 and will be removed in 0.39.0"
+        assert not memory_efficient_backward, "memory_efficient_backward is no longer required and the argument is deprecated in 0.37.0 and will be removed in 0.39.0"
         self.state = bnb.MatmulLtState()
         self.index = index
 

server/lorax_server/utils/tokens.py

@@ -1,9 +1,9 @@
 import re
+import warnings
 from contextlib import nullcontext
 from typing import List, Optional, Set, Tuple, Union
 
 import torch
-import warnings
 from transformers import (
     PreTrainedTokenizerBase,
     RepetitionPenaltyLogitsProcessor,