lint fixes

tests/dfencoder/test_scalers.py

@@ -115,8 +115,7 @@ def test_modified_scaler_transform(modified_scaler, tensor):
     assert torch.equal(torch.round(results, decimals=2), expected), f"{results} != {expected}"
 
     # Test alternate path where median absolute deviation is 1
-    t = torch.tensor([3.0, 4.0, 4.0, 5.0])
-    modified_scaler.fit(t)
+    modified_scaler.fit(torch.tensor([3.0, 4.0, 4.0, 5.0]))
     results = modified_scaler.transform(tensor)
     expected = torch.tensor([5.43, 6.86, 8.78])
     assert torch.equal(torch.round(results, decimals=2), expected), f"{results} != {expected}"
@@ -128,8 +127,7 @@ def test_modified_scaler_inverse_transform(modified_scaler, tensor):
     assert torch.equal(torch.round(results, decimals=2), expected), f"{results} != {expected}"
 
     # Test alternate path where median absolute deviation is 1
-    t = torch.tensor([3.0, 4.0, 4.0, 5.0])
-    modified_scaler.fit(t)
+    modified_scaler.fit(torch.tensor([3.0, 4.0, 4.0, 5.0]))
     results = modified_scaler.inverse_transform(tensor)
     expected = torch.tensor([8.64, 9.2, 9.95])
     assert torch.equal(torch.round(results, decimals=2), expected), f"{results} != {expected}"
@@ -161,13 +159,13 @@ def test_gauss_rank_scaler_fit_transform(gauss_rank_scaler, tensor):
 
 def test_null_scaler(tensor):
     orig = tensor.to(dtype=torch.float32, copy=True)
-    ns = scalers.NullScaler()
-    ns.fit(tensor)
+    scalar = scalers.NullScaler()
+    scalar.fit(tensor)
 
     # Verify it does nothing
-    assert ns.transform(tensor) is tensor
-    assert ns.inverse_transform(tensor) is tensor
-    assert ns.fit_transform(tensor) is tensor
+    assert scalar.transform(tensor) is tensor
+    assert scalar.inverse_transform(tensor) is tensor
+    assert scalar.fit_transform(tensor) is tensor
 
     # After all that the values should be the same
     assert torch.equal(tensor, orig), f"{tensor} != {orig}"

tests/llm/conftest.py

@@ -108,7 +108,7 @@ def mock_nemollm_fixture(mock_nemollm: mock.MagicMock):
     async def mock_task(fut: asyncio.Future, value: typing.Any = mock.DEFAULT):
         fut.set_result(value)
 
-    def create_future(*args, **kwargs) -> asyncio.Future:
+    def create_future(*args, **kwargs) -> asyncio.Future:  # pylint: disable=unused-argument
         event_loop = asyncio.get_event_loop()
         fut = event_loop.create_future()
         event_loop.create_task(mock_task(fut, mock.DEFAULT))