style(examples): standardize formatting

Tim Joseph · Tim Joseph · commit f47687c857fb · 2025-08-21T13:57:52.000+02:00
diff --git a/examples/tensor_dataclass/01_basic.py b/examples/tensor_dataclass/01_basic.py
@@ -32,7 +32,7 @@ def main() -> None:
     act_tensor = torch.rand(2)
 
     # Construct instance with required shape and device
-    data = SimpleData(
+    SimpleData(
         observations=obs_tensor,
         actions=act_tensor,
         shape=(2,),  # Batch dimensions shared by all fields
@@ -52,5 +52,6 @@ def main() -> None:
         # An error is expected here due to the shape mismatch.
         print(e)
 
+
 if __name__ == "__main__":
     main()
diff --git a/examples/tensor_dataclass/02_indexing.py b/examples/tensor_dataclass/02_indexing.py
@@ -52,6 +52,6 @@ def main() -> None:
     )
     data[1:3] = replacement
 
+
 if __name__ == "__main__":
     main()
-
diff --git a/examples/tensor_dataclass/08_detach_gradients.py b/examples/tensor_dataclass/08_detach_gradients.py
@@ -40,5 +40,6 @@ def main() -> None:
     assert not detached_batch.observations.requires_grad
     assert not detached_batch.actions.requires_grad
 
+
 if __name__ == "__main__":
-    main()
+    main()
diff --git a/examples/tensor_dataclass/09_nested.py b/examples/tensor_dataclass/09_nested.py
@@ -51,4 +51,4 @@ def main() -> None:
 
 
 if __name__ == "__main__":
-    main()
+    main()
diff --git a/examples/tensor_dict/01_basic.py b/examples/tensor_dict/01_basic.py
@@ -41,4 +41,4 @@ def main() -> None:
 
 
 if __name__ == "__main__":
-    main()
+    main()
diff --git a/examples/tensor_dict/02_indexing.py b/examples/tensor_dict/02_indexing.py
@@ -18,7 +18,7 @@ def main() -> None:
         shape=(6,),
         device="cpu",
     )
-    
+
     # Index a single item - removes batch dimension
     single_item = data[0]
     assert single_item.shape == ()
@@ -49,4 +49,4 @@ def main() -> None:
 
 
 if __name__ == "__main__":
-    main()
+    main()
diff --git a/examples/tensor_dict/03_shape_ops.py b/examples/tensor_dict/03_shape_ops.py
@@ -37,4 +37,4 @@ def main() -> None:
 
 
 if __name__ == "__main__":
-    main()
+    main()
diff --git a/examples/tensor_dict/05_device.py b/examples/tensor_dict/05_device.py
@@ -36,4 +36,4 @@ def main() -> None:
 
 
 if __name__ == "__main__":
-    main()
+    main()
diff --git a/examples/tensor_dict/07_copy_clone.py b/examples/tensor_dict/07_copy_clone.py
@@ -19,14 +19,14 @@ def main() -> None:
 
     # Shallow copy shares tensor memory
     shallow_copy = copy.copy(original)
-    assert shallow_copy["x"] is original["x"] # Same tensor objects
+    assert shallow_copy["x"] is original["x"]  # Same tensor objects
     assert shallow_copy["y"] is original["y"]
 
     # Clone creates independent tensors
     cloned = original.clone()
-    assert cloned["x"] is not original["x"] # Different tensor objects
+    assert cloned["x"] is not original["x"]  # Different tensor objects
     assert cloned["y"] is not original["y"]
 
 
 if __name__ == "__main__":
-    main()
+    main()
diff --git a/examples/tensor_dict/08_detach_gradients.py b/examples/tensor_dict/08_detach_gradients.py
@@ -39,4 +39,4 @@ def main() -> None:
 
 
 if __name__ == "__main__":
-    main()
+    main()
diff --git a/examples/tensor_dict/09_nested.py b/examples/tensor_dict/09_nested.py
@@ -28,7 +28,7 @@ def main() -> None:
         shape=(4,),
         device="cpu",
     )
-    
+
     full = TensorDict(
         {
             "env_state": torch.randn(4, 10),
@@ -47,4 +47,4 @@ def main() -> None:
 
 
 if __name__ == "__main__":
-    main()
+    main()
diff --git a/examples/tensor_dict/10_dynamic_keys.py b/examples/tensor_dict/10_dynamic_keys.py
@@ -15,7 +15,7 @@ def main() -> None:
         shape=(4,),
         device="cpu",
     )
-    
+
     # Add keys dynamically
     data["actions"] = torch.randn(4, 3)
     data["rewards"] = torch.randn(4)
@@ -26,10 +26,12 @@ def main() -> None:
 
     # Attribute shapes must be compatible with TensorDict shape
     try:
-        data["wrong_shape"] = torch.randn(3, 2)  # Wrong batch shape (3,) instead of (4,)
+        data["wrong_shape"] = torch.randn(
+            3, 2
+        )  # Wrong batch shape (3,) instead of (4,)
     except Exception as e:
         print(e)
 
 
 if __name__ == "__main__":
-    main()
+    main()
diff --git a/examples/tensor_dict/11_mapping_interface.py b/examples/tensor_dict/11_mapping_interface.py
@@ -27,8 +27,7 @@ def main() -> None:
 
     # Update from another TensorDict
     z = torch.randn(3, 1)
-    other = TensorDict({"z": z}, shape=(3,), device="cpu"
-                       )
+    other = TensorDict({"z": z}, shape=(3,), device="cpu")
     data.update(other)
     assert data["z"] is z
 
diff --git a/examples/tensor_dict/12_nested_dict_handling.py b/examples/tensor_dict/12_nested_dict_handling.py
@@ -17,7 +17,7 @@ def main() -> None:
                 "observations": torch.randn(3, 10),
                 "info": {
                     "step_count": torch.tensor([10, 15, 20]),
-                }
+                },
             },
             "agent": {
                 "policy": torch.randn(3, 6),
@@ -37,7 +37,7 @@ def main() -> None:
     # Access nested data
     assert data["env"]["observations"].shape == (3, 10)
     assert data["env"]["info"]["step_count"].shape == (3,)
-    
+
     # Operations propagate through automatic nesting
     reshaped = data.reshape(1, 3)
     assert reshaped["env"]["info"]["step_count"].shape == (1, 3)
@@ -57,4 +57,4 @@ def main() -> None:
 
 
 if __name__ == "__main__":
-    main()
+    main()
diff --git a/examples/tensor_dict/13_flatten_keys.py b/examples/tensor_dict/13_flatten_keys.py
@@ -27,11 +27,11 @@ def main() -> None:
     assert "env.obs" in flattened
     assert "env.info.step" in flattened
     assert "agent.policy" in flattened
-    
+
     # Memory is shared between original and flattened
     assert nested["env"]["obs"] is flattened["env.obs"]
     assert torch.equal(nested["env"]["info"]["step"], flattened["env.info.step"])
 
 
 if __name__ == "__main__":
-    main()
+    main()

Original file line number	Diff line number	Diff line change
`@@ -52,6 +52,6 @@ def main() -> None:`
`52`	`52`	`)`
`53`	`53`	`data[1:3] = replacement`
`54`	`54`
	`55`	`+`
`55`	`56`	`if __name__ == "__main__":`
`56`	`57`	`main()`
`57`		`-`
Original file line number	Diff line number	Diff line change
`@@ -51,4 +51,4 @@ def main() -> None:`
`51`	`51`
`52`	`52`
`53`	`53`	`if __name__ == "__main__":`
`54`		`- main()`
	`54`	`+ main()`
Original file line number	Diff line number	Diff line change
`@@ -41,4 +41,4 @@ def main() -> None:`
`41`	`41`
`42`	`42`
`43`	`43`	`if __name__ == "__main__":`
`44`		`- main()`
	`44`	`+ main()`
Original file line number	Diff line number	Diff line change
`@@ -37,4 +37,4 @@ def main() -> None:`
`37`	`37`
`38`	`38`
`39`	`39`	`if __name__ == "__main__":`
`40`		`- main()`
	`40`	`+ main()`
Original file line number	Diff line number	Diff line change
`@@ -36,4 +36,4 @@ def main() -> None:`
`36`	`36`
`37`	`37`
`38`	`38`	`if __name__ == "__main__":`
`39`		`- main()`
	`39`	`+ main()`
Original file line number	Diff line number	Diff line change
`@@ -39,4 +39,4 @@ def main() -> None:`
`39`	`39`
`40`	`40`
`41`	`41`	`if __name__ == "__main__":`
`42`		`- main()`
	`42`	`+ main()`