Fixed formatting

Author: Tim Joseph

tests/tensor_distribution/test_independent.py

@@ -41,7 +41,7 @@ def test_broadcasting_shapes(
         )
         assert td_independent.batch_shape == expected_batch_shape
         assert td_independent.dist().batch_shape == expected_batch_shape
-        
+
         # Test that shape property matches expected_batch_shape (fixes bug with reinterpreted_batch_ndims=0)
         assert td_independent.shape == expected_batch_shape
 

tests/tensor_distribution/test_tanh_normal.py

@@ -29,13 +29,15 @@ def test_initialization_tensor(self):
 
     def test_initialization_reinterpreted_batch_ndims(self):
         # Test with TensorIndependent wrapper for reinterpreted batch dimensions
-        from src.tensorcontainer.tensor_distribution.independent import TensorIndependent
-        
+        from src.tensorcontainer.tensor_distribution.independent import (
+            TensorIndependent,
+        )
+
         loc = torch.tensor([[0.0, 1.0], [2.0, 3.0]])
         scale = torch.tensor([[1.0, 2.0], [3.0, 4.0]])
         base_dist = TensorTanhNormal(loc, scale)
         dist = TensorIndependent(base_dist, reinterpreted_batch_ndims=1)
-        
+
         assert torch.equal(base_dist.loc, loc)
         assert torch.equal(base_dist.scale, scale)
         assert base_dist.shape == loc.shape