black

test/__init__.py

@@ -1,2 +1,2 @@
 # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
-# SPDX-License-Identifier: Apache-2.0
+# SPDX-License-Identifier: Apache-2.0

test/test_utils.py

@@ -4,10 +4,13 @@
 from chronos.utils import left_pad_and_stack_1D
 
 
-@pytest.mark.parametrize("tensors", [
-    list(map(torch.tensor, [[1, 2, 3], [5, 6]])),
-    list(map(torch.tensor, [[2.0, 3.0], [4.0, 5.0, 6.0], [7.0, 8.0, 9.0, 10.0]])),
-])
+@pytest.mark.parametrize(
+    "tensors",
+    [
+        list(map(torch.tensor, [[1, 2, 3], [5, 6]])),
+        list(map(torch.tensor, [[2.0, 3.0], [4.0, 5.0, 6.0], [7.0, 8.0, 9.0, 10.0]])),
+    ],
+)
 def test_pad_and_stack(tensors: list):
     stacked_and_padded = left_pad_and_stack_1D(tensors)
     assert stacked_and_padded.shape == (len(tensors), max(len(t) for t in tensors))

test/util.py

@@ -10,4 +10,4 @@ def validate_tensor(
     assert a.shape == shape
 
     if dtype is not None:
-        assert a.dtype == dtype
+        assert a.dtype == dtype