Update

[ghstack-poisoned]

Author: Vincent Moens

check_future_imports.py

@@ -0,0 +1,33 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+#
+# This source code is licensed under the MIT license found in the
+# LICENSE file in the root directory of this source tree.
+
+import os
+
+
+def find_files_without_future_annotations(directory):
+    """Finds Python files that do not contain 'from __future__ import annotations'."""
+    files_without_annotations = []
+
+    for root, _, files in os.walk(directory):
+        for file in files:
+            if file.endswith(".py"):
+                file_path = os.path.join(root, file)
+                with open(file_path, encoding="utf-8") as f:
+                    content = f.read()
+                    if "from __future__ import annotations" not in content:
+                        files_without_annotations.append(file_path)
+
+    return files_without_annotations
+
+
+if __name__ == "__main__":
+    repo_directory = "."  # Change this to your repository's root directory
+    files = find_files_without_future_annotations(repo_directory)
+    if files:
+        print("Files without 'from __future__ import annotations':")
+        for file in files:
+            print(file)
+    else:
+        print("All files contain 'from __future__ import annotations'.")

test/conftest.py

@@ -2,6 +2,8 @@
 #
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
+from __future__ import annotations
+
 import functools
 import os
 import sys

test/test_exploration.py

@@ -379,9 +379,8 @@ def test_nested(
         )
 
         action_spec = env.action_spec
-        d_act = action_spec.shape[-1]
+        action_spec.shape[-1]
 
-        nn.LazyLinear(d_act).to(device)
         policy = TensorDictModule(
             CountingEnvCountModule(action_spec=action_spec),
             in_keys=[("data", "states") if nested_obs_action else "observation"],

test/test_transforms.py

@@ -4785,7 +4785,7 @@ def make_env():
             )
             return env
 
-        SerialEnv(2, make_env)
+        SerialEnv(2, make_env).check_env_specs()
 
     def test_parallel_trans_env_check(self, maybe_fork_ParallelEnv):
         def make_env():

torchrl/_extension.py

@@ -2,6 +2,7 @@
 #
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
+from __future__ import annotations
 
 import importlib.util
 import warnings

torchrl/collectors/distributed/default_configs.py

@@ -1,3 +1,9 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+#
+# This source code is licensed under the MIT license found in the
+# LICENSE file in the root directory of this source tree.
+from __future__ import annotations
+
 import os
 
 TCP_PORT = os.environ.get("TCP_PORT", "10003")

torchrl/collectors/distributed/utils.py

@@ -1,3 +1,10 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+#
+# This source code is licensed under the MIT license found in the
+# LICENSE file in the root directory of this source tree.
+
+from __future__ import annotations
+
 import subprocess
 import time
 

torchrl/data/datasets/d4rl_infos.py

@@ -2,6 +2,7 @@
 #
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
+from __future__ import annotations
 
 D4RL_DATASETS = {
     "maze2d-open-v0": "http://rail.eecs.berkeley.edu/datasets/offline_rl/maze2d/maze2d-open-sparse.hdf5",

torchrl/data/datasets/utils.py

@@ -2,6 +2,8 @@
 #
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
+from __future__ import annotations
+
 import os
 
 

torchrl/envs/custom/pendulum.py

@@ -2,6 +2,8 @@
 #
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
+from __future__ import annotations
+
 import numpy as np
 
 import torch